def test_no_target_datapath(self):
scheme = AKScheme(gamma=1)
secret_key = 123
recipient = 0
data = '../../datasets/breast_cancer_wisconsin.csv'
fingerprinted_data = scheme.insertion(data, recipient, secret_key)
suspect = scheme.detection(fingerprinted_data, secret_key)
self.assertEqual(recipient, suspect)
def test_detection_all_numerical(self):
scheme = AKScheme(gamma=2)
secret_key = 123
recipient = 0
data = BreastCancerWisconsin()
fingerprinted_data = scheme.insertion(data, recipient, secret_key)
suspect = scheme.detection(fingerprinted_data, secret_key)
self.assertEqual(recipient, suspect)
def test_include_fail(self):
scheme = AKScheme(gamma=1)
secret_key = 123
recipient = 0
data = '../../datasets/breast_cancer_wisconsin.csv'
dataframe = pd.read_csv(data)
include = ['clump-thickness', 'mitoses']
fingerprinted_data = scheme.insertion(dataset=dataframe,
recipient_id=recipient,
secret_key=secret_key,
include=include)
suspect = scheme.detection(fingerprinted_data, secret_key)
self.assertNotEqual(recipient, suspect)
def test_include_1(self):
scheme = AKScheme(gamma=1)
secret_key = 123
recipient = 0
data = '../../datasets/breast_cancer_wisconsin.csv'
dataframe = pd.read_csv(data)
fingerprinted_data = scheme.insertion(dataset=dataframe,
recipient_id=recipient,
secret_key=secret_key,
include=['clump-thickness'])
suspect = scheme.detection(fingerprinted_data,
secret_key,
include=['clump-thickness'])
self.assertFalse(
fingerprinted_data.dataframe['clump-thickness'].equals(
dataframe['clump-thickness']))
def test_exclude_multi(self):
scheme = AKScheme(gamma=1)
secret_key = 123
recipient = 0
data = '../../datasets/breast_cancer_wisconsin.csv'
dataframe = pd.read_csv(data)
exclude = [
'clump-thickness', 'uniformity-of-cell-shape', 'marginal-adhesion',
'single-epithelial-cell-size', 'bare-nuclei', 'bland-chromatin',
'sample-code-number', 'normal-nucleoli'
]
fingerprinted_data = scheme.insertion(dataframe,
recipient,
secret_key,
exclude=exclude)
suspect = scheme.detection(fingerprinted_data,
secret_key,
exclude=exclude)
self.assertEqual(recipient, suspect)
