diff options
-rw-r--r-- | tensorflow/python/kernel_tests/control_flow_ops_py_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py index 68873df97e..b567b71424 100644 --- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py +++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py @@ -734,11 +734,11 @@ class ControlFlowTest(test.TestCase): def body_fn(i): with ops.control_dependencies([increment]): - return i + i + return i + 1 - result = control_flow_ops.while_loop(cond=lambda i: i < 1, + result = control_flow_ops.while_loop(cond=lambda i: i < 2, body=body_fn, loop_vars=[1]) - result.eval() + self.assertAllEqual(result.eval(), 2) self.assertAllEqual(v.eval(), 1.0) def testWhileExternalControlDependenciesNoInput(self): |