test: add almmost_equal assert

jina-ai · Sep 30, 2020 · ee710ec · ee710ec
1 parent 04e1de6
commit ee710ec
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/unit/flow/flow.jpg b/tests/unit/flow/flow.jpg
diff --git a/tests/unit/flow/test_flow_visualization.py b/tests/unit/flow/test_flow_visualization.py
@@ -1,4 +1,6 @@
 import os
+import numpy as np
+
 from PIL import Image
 from jina.flow import Flow
 
@@ -14,3 +16,4 @@ def test_visualization():
     flow_original = Image.open(os.path.join(cur_dir, 'flow_original.jpg'))
     flow_created = Image.open(os.path.join(cur_dir, 'flow.jpg'))
     assert flow_original.size == flow_created.size
+    np.testing.assert_array_almost_equal(flow_original, flow_created)