diff options
author | 2016-12-29 22:46:24 -0800 | |
---|---|---|
committer | 2016-12-29 23:06:59 -0800 | |
commit | e121667dc609de978a223c56ee906368d2c4ceef (patch) | |
tree | 7d4e1f1e1b4fd469487872c0cd34ddace5ac570c /tensorflow/contrib/tensorboard/plugins/trace/trace_test.py | |
parent | 7815fcba7767aa1eb3196c5861e174f8b3c43bab (diff) |
Remove so many more hourglass imports
Change: 143230429
Diffstat (limited to 'tensorflow/contrib/tensorboard/plugins/trace/trace_test.py')
-rw-r--r-- | tensorflow/contrib/tensorboard/plugins/trace/trace_test.py | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/tensorflow/contrib/tensorboard/plugins/trace/trace_test.py b/tensorflow/contrib/tensorboard/plugins/trace/trace_test.py index e67bde9d59..d44f46b5bb 100644 --- a/tensorflow/contrib/tensorboard/plugins/trace/trace_test.py +++ b/tensorflow/contrib/tensorboard/plugins/trace/trace_test.py @@ -19,27 +19,30 @@ from __future__ import division from __future__ import print_function import tempfile -import tensorflow as tf from google.protobuf import json_format + from tensorflow.contrib.tensorboard.plugins import trace +from tensorflow.python.framework import constant_op +from tensorflow.python.platform import gfile +from tensorflow.python.platform import test -class TraceTest(tf.test.TestCase): +class TraceTest(test.TestCase): def setUp(self): self._temp_dir = tempfile.mkdtemp() self._temp_trace_json = self._temp_dir + 'trace.json' def tearDown(self): - tf.gfile.DeleteRecursively(self._temp_dir) + gfile.DeleteRecursively(self._temp_dir) def testEmptyGraph(self): trace_info = self._store_and_read_trace_info() self.assertEqual(len(trace_info.ops), 0) def testHasSourceCodeOfThisFile(self): - tf.constant(0) + constant_op.constant(0) trace_info = self._store_and_read_trace_info() self.assertTrue(trace_info.files) @@ -49,7 +52,7 @@ class TraceTest(tf.test.TestCase): self.fail('trace_test file not found in the trace info json') def testHasTheConstantOp(self): - tf.constant(0) + constant_op.constant(0) trace_info = self._store_and_read_trace_info() self.assertTrue(trace_info.ops) @@ -81,11 +84,12 @@ class TraceTest(tf.test.TestCase): trace.store_trace_info(self._temp_trace_json) trace_info = trace.TraceInfo() - with tf.gfile.Open(self._temp_trace_json) as f: + with gfile.Open(self._temp_trace_json) as f: text = f.read().decode('utf-8') json_format.Parse(text, trace_info) return trace_info + if __name__ == '__main__': - tf.test.main() + test.main() |