experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test case for error message of Dimension div (/)

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2019-02-18 00:36:06 +00:00
parent 7e5d6c25f4
commit 1c241df19f
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/python/framework/tensor_shape_test.py
View File

@ -205,6 +205,23 @@ class DimensionTest(test_util.TensorFlowTestCase):
reconstructed = ctor(*args) reconstructed = ctor(*args)
self.assertEquals(reconstructed, dim) 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): class ShapeTest(test_util.TensorFlowTestCase):