diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-30 17:01:47 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-30 17:05:56 -0700 |
| commit | 8fee2e4b7c915d952332dc8cc9be7cfefea35162 (patch) | |
| tree | 01c2fd830cbc01819470fdc5c667cb35d50fc40f /tensorflow/python/kernel_tests/control_flow_ops_py_test.py | |
| parent | 934b722bff332f51973d9a9d0f65c69a51292406 (diff) | |
Update control loop test, so we don't depend on a control dependency in the body of a loop executed zero times.
PiperOrigin-RevId: 206672040
Diffstat (limited to 'tensorflow/python/kernel_tests/control_flow_ops_py_test.py')
| -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): |