From 1c241df19ffbaf562ea51d67bb167245dd6b7b95 Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Mon, 18 Feb 2019 00:36:06 +0000 Subject: [PATCH] Add test case for error message of Dimension div (/) Signed-off-by: Yong Tang --- .../python/framework/tensor_shape_test.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tensorflow/python/framework/tensor_shape_test.py b/tensorflow/python/framework/tensor_shape_test.py index 7d85e0a99e6..786e4ef762a 100644 --- a/tensorflow/python/framework/tensor_shape_test.py +++ b/tensorflow/python/framework/tensor_shape_test.py @@ -205,6 +205,23 @@ class DimensionTest(test_util.TensorFlowTestCase): reconstructed = ctor(*args) self.assertEquals(reconstructed, dim) + def testDiv(self): + # Note: This test is related to GitHub issue 25790. + six = tensor_shape.Dimension(6) + two = tensor_shape.Dimension(2) + message = (r"unsupported operand type\(s\) for \/: " + r"'Dimension' and 'Dimension', please use \/\/ instead") + with self.assertRaisesRegexp(TypeError, message): + _ = six / two + message = (r"unsupported operand type\(s\) for \/: " + r"'Dimension' and 'int', please use \/\/ instead") + with self.assertRaisesRegexp(TypeError, message): + _ = six / 2 + message = (r"unsupported operand type\(s\) for \/: " + r"'int' and 'Dimension', please use \/\/ instead") + with self.assertRaisesRegexp(TypeError, message): + _ = 6 / two + class ShapeTest(test_util.TensorFlowTestCase):