test: add more points in validating flip.

Author: stevenhua0320

tests/test_load_image.py

@@ -1,7 +1,7 @@
 import os
 import shutil
 from pathlib import Path
-from unittest.mock import Mock
+from types import SimpleNamespace
 
 import pytest
 
@@ -12,30 +12,30 @@
 load_image_param = [
     # case 1: just filename of file in current directory.
     # expect function loads tiff file from cwd
-    ("KFe2As2-00838.tif", Mock(fliphorizontal=False, flipvertical=False), 0),
+    (["KFe2As2-00838.tif", False, False], [0, 26, 173]),
     # case 2: absolute file path to file in another directory.
     # expect file is found and correctly read.
     (
-        "home_dir/KFe2As2-00838.tif",
-        Mock(fliphorizontal=True, flipvertical=False),
-        102,
+        ["home_dir/KFe2As2-00838.tif", True, False],
+        [102, 57, 136],
     ),
     # case 3: relative file path to file in another directory.
     # expect file is found and correctly read
-    ("./KFe2As2-00838.tif", Mock(fliphorizontal=False, flipvertical=True), 39),
+    (["./KFe2As2-00838.tif", False, True], [39, 7, 0]),
     # case 4: non-existent file that incurred by mistype.
     (
-        "nonexistent_file.tif",
-        Mock(fliphorizontal=False, flipvertical=False),
+        ["nonexistent_file.tif", False, False],
         FileNotFoundError,
     ),
+    # case 5: relative file path to file in another directory.
+    # expect file to be flip both horizontally and vertically
+    # and correctly read
+    (["./KFe2As2-00838.tif", True, True], [0, 53, 21]),
 ]
 
 
-@pytest.mark.parametrize(
-    "file_name, config, expected_entry_value", load_image_param
-)
-def test_load_image(file_name, config, expected_entry_value, user_filesystem):
+@pytest.mark.parametrize("inputs, expected", load_image_param)
+def test_load_image(inputs, expected, user_filesystem):
     home_dir = user_filesystem["home"]
     cwd_dir = user_filesystem["cwd"]
     os.chdir(cwd_dir)
@@ -49,13 +49,15 @@ def test_load_image(file_name, config, expected_entry_value, user_filesystem):
     )
     shutil.copy(source_file, cwd_dir / "KFe2As2-00838.tif")
     shutil.copy(source_file, home_dir / "KFe2As2-00838.tif")
-
+    config = SimpleNamespace(fliphorizontal=inputs[1], flipvertical=inputs[2])
     try:
         loader = LoadImage(config)
-        actual = loader.load_image(file_name)
+        actual = loader.load_image(inputs[0])
         assert actual.shape == expected_shape
         assert actual.mean() == expected_mean
-        assert actual[1][0] == expected_entry_value
+        assert actual[1][0] == expected[0]
+        assert actual[1][1] == expected[1]
+        assert actual[2][5] == expected[2]
     except FileNotFoundError:
         pytest.raises(
             FileNotFoundError,