python source code of test

Project: differential-privacy-library (GitHub Link)

differential-privacy-library-master
- .coveragerc
- readthedocs.yml
- diffprivlib
  - models
    - k_means.py
    - linear_regression.py
    - standard_scaler.py
    - __init__.py
    - naive_bayes.py
    - pca.py
    - logistic_regression.py
  - validation.py
  - mechanisms
    - vector.py
    - binary.py
    - geometric.py
    - uniform.py
    - wishart.py
    - exponential.py
    - laplace.py
    - __init__.py
    - staircase.py
    - transforms
      - stringtoint.py
      - __init__.py
      - roundedinteger.py
      - base.py
      - inttostring.py
    - gaussian.py
    - base.py
  - __init__.py
  - utils.py
  - accountant.py
  - tools
    - histograms.py
    - __init__.py
    - utils.py
- .pylintrc
- notebooks
  - pipeline_accuracy_500.p
  - logistic_regression.ipynb
  - pipeline.ipynb
  - naive_bayes.ipynb
  - linear_regression.ipynb
  - lr_accuracy_500.p
  - 30seconds.ipynb
  - accountant.ipynb
  - histograms.ipynb
  - exploration.ipynb
- setup.py
- LICENSE.md
- setup.cfg
- .travis.yml
- README.md
- tests
  - test_BudgetAccountant.py
  - models
    - test_KMeans.py
    - test_incremental_mean_and_var.py
    - test_LinearRegression.py
    - test_LogisticRegression.py
    - test_StandardScaler.py
    - test_logistic_regression_path.py
    - __init__.py
    - test_GaussianNB.py
    - test_PCA.py
  - test_Budget.py
  - mechanisms
    - test_GeometricFolded.py
    - test_DPMechanism.py
    - test_GeometricTruncated.py
    - test_LaplaceFolded.py
    - test_Laplace.py
    - test_Staircase.py
    - test_TruncationAndFoldingMixin.py
    - test_ExponentialHierarchical.py
    - test_Binary.py
    - test_DPMachine.py
    - test_Gaussian.py
    - test_Exponential.py
    - test_GaussianAnalytic.py
    - test_LaplaceBoundedDomain.py
    - test_LaplaceTruncated.py
    - __init__.py
    - test_Uniform.py
    - test_LaplaceBoundedNoise.py
    - test_Wishart.py
    - test_Vector.py
    - transforms
      - test_IntToString.py
      - test_StringToInt.py
      - test_DPTransformer.py
      - test_RoundedInteger.py
      - __init__.py
    - test_GaussianDiscrete.py
    - test_Geometric.py
  - test_clip_to_bounds.py
  - test_check_bounds.py
  - __init__.py
  - test_check_epsilon_delta.py
  - tools
    - test_histogram2d.py
    - test_nanmean.py
    - test_var.py
    - test_histogram.py
    - test_histogramdd.py
    - test_count_nonzero.py
    - test_mean.py
    - __init__.py
    - test_nansum.py
    - test_nanvar.py
    - test_sum.py
    - test_nanstd.py
    - test_std.py
  - test_clip_to_norm.py
- .gitignore
- docs
  - Makefile
  - make.bat
  - index.rst
  - conf.py
  - modules
    - mechanisms.rst
    - accountant.rst
    - transforms.rst
    - utils.rst
    - tools.rst
    - models.rst
    - validation.rst

import abc
from unittest import TestCase

from numpy.random import random

from diffprivlib.mechanisms import DPMachine


class TestDPMachine(TestCase):
    def test_not_none(self):
        self.assertIsNotNone(DPMachine)

    def test_class(self):
        self.assertTrue(issubclass(DPMachine, abc.ABC))

    def test_instantiation(self):
        with self.assertRaises(TypeError):
            DPMachine()

    def test_base(self):
        class BaseDPMachine(DPMachine):
            def set_epsilon(self, epsilon):
                return self

            def set_epsilon_delta(self, epsilon, delta):
                return self

            def randomise(self, value):
                return random()

        mech = BaseDPMachine()
        val = mech.randomise(1)

        self.assertIsInstance(val, float)