626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Internal change.

Browse source 
PiperOrigin-RevId: 321768596
This commit is contained in:
A. Unique TensorFlower 2020-07-17 06:37:23 -07:00
parent 3549d23da3
commit 8ec709e3d7
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tensorflow_privacy/privacy/membership_inference_attack/data_structures_test.py
View file

@ -44,6 +44,22 @@ class AttackInputDataTest(absltest.TestCase):
np.testing.assert_equal( np.testing.assert_equal(
attack_input.get_loss_test().tolist(), [1.0, 4.0, 6.0]) attack_input.get_loss_test().tolist(), [1.0, 4.0, 6.0])
def test_validator(self):
self.assertRaises(ValueError,
AttackInputData(logits_train=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData(labels_train=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData(loss_train=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData(logits_test=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData(labels_test=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData(loss_test=np.array([])).validate)
self.assertRaises(ValueError,
AttackInputData().validate)
if __name__ == '__main__': if __name__ == '__main__':
absltest.main() absltest.main()