diff options
Diffstat (limited to 'tensorflow/python/debug/cli/analyzer_cli_test.py')
-rw-r--r-- | tensorflow/python/debug/cli/analyzer_cli_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/debug/cli/analyzer_cli_test.py b/tensorflow/python/debug/cli/analyzer_cli_test.py index d6b5c3a097..5d3f1aed2f 100644 --- a/tensorflow/python/debug/cli/analyzer_cli_test.py +++ b/tensorflow/python/debug/cli/analyzer_cli_test.py @@ -1608,7 +1608,8 @@ class AnalyzerCLIWhileLoopTest(test_util.TensorFlowTestCase): self.assertEqual(" dtype: int32", output.lines[1]) self.assertEqual(" shape: ()", output.lines[2]) self.assertEqual("", output.lines[3]) - self.assertEqual("array(%d, dtype=int32)" % i, output.lines[4]) + self.assertTrue(output.lines[4].startswith("array(%d" % i)) + self.assertTrue(output.lines[4].endswith(")")) def testMultipleDumpsPrintTensorInvalidNumber(self): output = self._registry.dispatch_command("pt", |