Skip to content

Commit

Permalink
tfdbg: Add test for loading DebugDumpDir with a relative path
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 174999937
  • Loading branch information
caisq authored and tensorflower-gardener committed Nov 8, 2017
1 parent af9c4ea commit 59ea341
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions tensorflow/python/debug/wrappers/dumping_wrapper_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,20 @@ def testDumpingOnASingleRunWorks(self):
self.assertEqual(repr(self.inc_v), dump.run_fetches_info)
self.assertEqual(repr(None), dump.run_feed_keys_info)

def testDumpingOnASingleRunWorksWithRelativePathForDebugDumpDir(self):
sess = dumping_wrapper.DumpingDebugWrapperSession(
self.sess, session_root=self.session_root, log_usage=False)
sess.run(self.inc_v)
dump_dirs = glob.glob(os.path.join(self.session_root, "run_*"))
cwd = os.getcwd()
try:
os.chdir(self.session_root)
dump = debug_data.DebugDumpDir(
os.path.relpath(dump_dirs[0], self.session_root))
self.assertAllClose([10.0], dump.get_tensors("v", 0, "DebugIdentity"))
finally:
os.chdir(cwd)

def testDumpingOnASingleRunWithFeedDictWorks(self):
sess = dumping_wrapper.DumpingDebugWrapperSession(
self.sess, session_root=self.session_root, log_usage=False)
Expand Down

0 comments on commit 59ea341

Please sign in to comment.