From 1080be335339ccef8a3384585c23219ec116dece Mon Sep 17 00:00:00 2001 From: wawltor Date: Fri, 21 Aug 2020 10:13:55 +0800 Subject: [PATCH] update the test_mean test case for bug fix update the test_mean test case --- python/paddle/fluid/tests/unittests/test_mean_op.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_mean_op.py b/python/paddle/fluid/tests/unittests/test_mean_op.py index a2befb4a29a..8faa8825838 100644 --- a/python/paddle/fluid/tests/unittests/test_mean_op.py +++ b/python/paddle/fluid/tests/unittests/test_mean_op.py @@ -101,7 +101,7 @@ class TestMeanAPI(unittest.TestCase): fetch_list=[out1, out2, out3, out4, out5]) out_ref = np.mean(self.x) for out in res: - self.assertEqual(np.allclose(out, out_ref), True) + self.assertEqual(np.allclose(out, out_ref, rtol=1e-04), True) def test_api_dygraph(self): paddle.disable_static(self.place) @@ -114,7 +114,9 @@ class TestMeanAPI(unittest.TestCase): if len(axis) == 0: axis = None out_ref = np.mean(x, axis, keepdims=keepdim) - self.assertEqual(np.allclose(out.numpy(), out_ref), True) + self.assertEqual( + np.allclose( + out.numpy(), out_ref, rtol=1e-04), True) test_case(self.x) test_case(self.x, []) -- GitLab