Skip to content

Commit 99d8d03

Browse files
Monica Songtensorflower-gardener
authored andcommitted
[SavedModel Fingerprinting] Add python tests for fingerprint #4.
RFC: tensorflow/community#415 PiperOrigin-RevId: 465618721
1 parent 42e425a commit 99d8d03

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow/python/saved_model/pywrap_saved_model_fingerprinting_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
class FingerprintingTest(test.TestCase):
2424

25-
def test_graphdef_basic(self):
25+
def test_module_basic(self):
2626
sm_pb_file = test.test_src_dir_path(
2727
"cc/saved_model/testdata/VarsAndArithmeticObjectGraph/saved_model.pb")
2828
with file_io.FileIO(sm_pb_file, "rb") as f:
@@ -37,6 +37,8 @@ def test_graphdef_basic(self):
3737
self.assertEqual(fingerprint_def.graph_def_program_hash,
3838
13188891313422428336)
3939
self.assertEqual(fingerprint_def.signature_def_hash, 5693392539583495303)
40+
self.assertEqual(fingerprint_def.saved_object_graph_hash,
41+
3678101440349108924)
4042

4143

4244
if __name__ == "__main__":

0 commit comments

Comments
 (0)