test: allow for another output in test_flowchart

Author: percevalw

eds_scikit/biology/utils/config.py

@@ -1,9 +1,9 @@
 import glob
+import os
 from pathlib import Path
 from typing import List
 
 import pandas as pd
-from importlib_metadata import os
 from loguru import logger
 
 from eds_scikit.biology.utils.process_concepts import ConceptsSet

eds_scikit/utils/test_utils.py

@@ -96,7 +96,7 @@ def assert_equal(
     return output
 
 
-def assert_images_equal(image_1: str, image_2: str):
+def images_are_equal(image_1: str, image_2: str):
     img1 = Image.open(image_1)
     img2 = Image.open(image_2)
 
@@ -110,7 +110,7 @@ def assert_images_equal(image_1: str, image_2: str):
 
     if sum_sq_diff == 0:
         # Images are exactly the same
-        pass
+        return True
     else:
         normalized_sum_sq_diff = sum_sq_diff / np.sqrt(sum_sq_diff)
-        assert normalized_sum_sq_diff < 0.001
+        return normalized_sum_sq_diff < 0.001

tests/flowchart/expected_flowchart_bis.png

@@ -5,7 +5,7 @@
 import pytest
 
 from eds_scikit.utils.flowchart import Flowchart
-from eds_scikit.utils.test_utils import assert_images_equal
+from eds_scikit.utils.test_utils import images_are_equal
 
 data_as_df = pd.DataFrame(
     dict(
@@ -63,12 +63,13 @@ def test_flowchart(data, tmpdir_factory):
 
     _ = F.generate_flowchart(alternate=True, fontsize=10)
 
-    result_path = tmp_dir / "flowchart.png"
-    F.save(result_path, dpi=72)
+    out_path = tmp_dir / "flowchart.png"
+    F.save(out_path, dpi=72)
 
-    expected = Path(__file__).parent / "expected_flowchart.png"
+    target_1 = Path(__file__).parent / "expected_flowchart.png"
+    target_2 = Path(__file__).parent / "expected_flowchart_bis.png"
 
-    assert_images_equal(result_path, expected)
+    assert images_are_equal(out_path, target_1) or images_are_equal(out_path, target_2)
 
 
 def test_incorrect_data():