diff options
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 | 7 |
1 files changed, 7 insertions, 0 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 58302a683d..6de4c905b1 100644 --- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py +++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py @@ -330,6 +330,13 @@ class ControlFlowTest(tf.test.TestCase): result = exit_i.eval() self.assertAllEqual(10, result) + def testCondBool(self): + values = tf.constant(10) + fn1 = lambda: tf.add(values, 1) + fn2 = lambda: tf.sub(values, 1) + with self.assertRaisesRegexp(TypeError, "must not be a Python bool"): + _ = control_flow_ops.cond(False, fn1, fn2) + def testCondIndexedSlices(self): with self.test_session(): values = tf.constant(10) |