diff options
author | Alexander Gorban <gorban@google.com> | 2017-12-12 20:09:45 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-12 20:12:52 -0800 |
commit | 216878ea3dafdc5fbe6a15d389edb003ad2fd4b4 (patch) | |
tree | 9607fd086c774caeeb96cd5d98219bc2b86e8cc5 /tensorflow/python/kernel_tests/control_flow_ops_py_test.py | |
parent | b7308e3bd69349e9023497948a6bf55d3b0895d9 (diff) |
Simplify tf.case implementation.
PiperOrigin-RevId: 178853258
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 | 4 |
1 files changed, 2 insertions, 2 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 35ae89ed33..5b0abaa2eb 100644 --- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py +++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py @@ -2279,8 +2279,7 @@ class ControlFlowTest(test.TestCase): # Duplicate events cause an error if exclusive = True r4 = control_flow_ops.case( [(x < y, f1), (x < y, f2)], default=f3, exclusive=True) - with self.assertRaisesOpError( - "More than one condition evaluated as True but exclusive=True."): + with self.assertRaisesOpError("Input error:"): r4.eval() # Check that the default is called if none of the others are @@ -3045,5 +3044,6 @@ class EagerTest(test.TestCase): default=f3, exclusive=True) self.assertAllEqual(r1.numpy(), 17) + if __name__ == "__main__": test.main() |