diff --git a/tensorflow/python/kernel_tests/division_future_test.py b/tensorflow/python/kernel_tests/division_future_test.py index 85c85809d3f..b0a5d8c7211 100644 --- a/tensorflow/python/kernel_tests/division_future_test.py +++ b/tensorflow/python/kernel_tests/division_future_test.py @@ -47,7 +47,7 @@ class DivisionTestCase(test.TestCase): tensors.append((x, y)) def f(x, y): self.assertEqual(x.dtype, y.dtype) - self.assertEqual(x, y) + self.assertAllClose(x, y) checks.append(f) with self.cached_session() as sess: diff --git a/tensorflow/python/kernel_tests/division_past_test.py b/tensorflow/python/kernel_tests/division_past_test.py index 38bb18631ab..7701edfa1b5 100644 --- a/tensorflow/python/kernel_tests/division_past_test.py +++ b/tensorflow/python/kernel_tests/division_past_test.py @@ -14,7 +14,7 @@ # ============================================================================== """Tests for division with division imported from __future__. -This file should be exactly the same as division_past_test.py except +This file should be exactly the same as division_future_test.py except for the __future__ division line. """ @@ -46,7 +46,7 @@ class DivisionTestCase(test.TestCase): tensors.append((x, y)) def f(x, y): self.assertEqual(x.dtype, y.dtype) - self.assertEqual(x, y) + self.assertAllClose(x, y) checks.append(f) with self.cached_session() as sess: