From 0e5089b87fbbac44cb2f8a3bdd4b658ca48b73b0 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Fri, 3 Nov 2023 11:53:08 +0800 Subject: [PATCH] use assertLess --- tests/test_flip_image.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_flip_image.py b/tests/test_flip_image.py index e247f33..9b0719a 100644 --- a/tests/test_flip_image.py +++ b/tests/test_flip_image.py @@ -36,9 +36,9 @@ class FlipImageTestCase(unittest.TestCase): delimiter=",", dtype=int, ) - self.assertEqual( - flip_image(image_input), - image_answer, + self.assertLess( + np.linalg.norm(flip_image(image_input) - image_answer), + 1e-6, "Test flip image on server failed", ) -- GitLab