diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/scalar_strict_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/scalar_strict_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/python/kernel_tests/scalar_strict_test.py b/tensorflow/python/kernel_tests/scalar_strict_test.py index 9b9babbdc1..e208217637 100644 --- a/tensorflow/python/kernel_tests/scalar_strict_test.py +++ b/tensorflow/python/kernel_tests/scalar_strict_test.py @@ -70,13 +70,13 @@ class ScalarStrictTest(test.TestCase): self.assertAllEqual(r, correct) def testConcat(self): - self.check(array_ops.concat_v2, (([2], [3], [7]), [0]), + self.check(array_ops.concat, (([2], [3], [7]), [0]), 'axis tensor should be a scalar integer', [2, 3, 7]) for data in (2, 3, 7), (2, [3], 7), (2, 3, [7]): - self.check(array_ops.concat_v2, (data, 0), + self.check(array_ops.concat, (data, 0), r'Expected \w+ dimensions in the range \[0, 0\)', [2, 3, 7]) for data in ([2], 3, 7), ([2], [3], 7): - self.check(array_ops.concat_v2, (data, 0), + self.check(array_ops.concat, (data, 0), r'Ranks of all input tensors should match', [2, 3, 7]) def testFill(self): |