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

Fix error with scalar values in assertAllInRange() input.

Browse Source 
PiperOrigin-RevId: 334882384
Change-Id: If1bf4e3da2df22460786b849416735054f482f38
This commit is contained in:
Dmitry Tsarkov 2020-10-01 13:14:13 -07:00 committed by TensorFlower Gardener
parent 8991d372bf
commit 2e69a2dc42
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/framework/test_util.py
View File

@ -2904,6 +2904,8 @@ class TensorFlowTestCase(googletest.TestCase):
lines = []
subscripts = np.transpose(subscripts)
prefix = " " * indent
if np.ndim(value) == 0:
return [prefix + "[0] : " + str(value)]
for subscript in itertools.islice(subscripts, limit):
lines.append(prefix + str(subscript) + " : " +
str(value[tuple(subscript)]))

13
tensorflow/python/framework/test_util_test.py
View File

@ -596,6 +596,19 @@ class TestUtilTest(test_util.TensorFlowTestCase, parameterized.TestCase):
self.assertAllInRange(
x, 10, 15, open_lower_bound=True, open_upper_bound=True)
@test_util.run_in_graph_and_eager_modes
def testAssertAllInRangeScalar(self):
x = constant_op.constant(10.0, name="x")
nan = constant_op.constant(np.nan, name="nan")
self.assertAllInRange(x, 5, 15)
with self.assertRaises(AssertionError):
self.assertAllInRange(nan, 5, 15)
with self.assertRaises(AssertionError):
self.assertAllInRange(x, 10, 15, open_lower_bound=True)
with self.assertRaises(AssertionError):
self.assertAllInRange(x, 1, 2)
@test_util.run_in_graph_and_eager_modes
def testAssertAllInRangeErrorMessageEllipses(self):
x_init = np.array([[10.0, 15.0]] * 12)