ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

Internal change.

Browse source 
PiperOrigin-RevId: 325249654
This commit is contained in:
A. Unique TensorFlower 2020-08-06 09:47:49 -07:00
parent 08f960a1af
commit efca03b593
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -14,6 +14,8 @@
# Lint as: python3
"""Tests for tensorflow_privacy.privacy.membership_inference_attack.data_structures."""
import os
import tempfile
from absl.testing import absltest
import numpy as np
from tensorflow_privacy.privacy.membership_inference_attack.data_structures import AttackInputData
@ -205,6 +207,17 @@ class AttackResultsTest(absltest.TestCase):
'Highest advantage on slice SingleSliceSpec(Entire dataset) achieved ' +
'by AttackType.THRESHOLD_ATTACK with an advantage of 0.0')
def test_save_load(self):
results = AttackResults(
[self.perfect_classifier_result, self.random_classifier_result])
with tempfile.TemporaryDirectory() as tmpdirname:
filepath = os.path.join(tmpdirname, 'results.pickle')
results.save(filepath)
loaded_results = AttackResults.load(filepath)
self.assertEqual(repr(results), repr(loaded_results))
if __name__ == '__main__':
absltest.main()