diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/regex_full_match_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/regex_full_match_op_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/python/kernel_tests/regex_full_match_op_test.py b/tensorflow/python/kernel_tests/regex_full_match_op_test.py index 7bd8c3ca27..e81f562a2a 100644 --- a/tensorflow/python/kernel_tests/regex_full_match_op_test.py +++ b/tensorflow/python/kernel_tests/regex_full_match_op_test.py @@ -35,7 +35,7 @@ class RegexFullMatchOpVariantsTest(test.TestCase, parameterized.TestCase): def testRegexFullMatch(self, op): values = ["abaaba", "abcdabcde"] - with self.test_session(): + with self.cached_session(): input_tensor = constant_op.constant(values, dtypes.string) matched = op(input_tensor, "a.*a").eval() self.assertAllEqual([True, False], matched) @@ -49,14 +49,14 @@ class RegexFullMatchOpVariantsTest(test.TestCase, parameterized.TestCase): def testEmptyMatch(self, op): values = ["abc", "1"] - with self.test_session(): + with self.cached_session(): input_tensor = constant_op.constant(values, dtypes.string) matched = op(input_tensor, "").eval() self.assertAllEqual([False, False], matched) def testInvalidPattern(self, op): values = ["abc", "1"] - with self.test_session(): + with self.cached_session(): input_tensor = constant_op.constant(values, dtypes.string) invalid_pattern = "A[" matched = op(input_tensor, invalid_pattern) |