diff --git a/tests/test_flip_image.py b/tests/test_flip_image.py index e247f337436ae3dd08b5c0f3552a0413ace83944..9b0719acd5ae34ee4eeb1be9b5e2bc28be90ec46 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", )