style: change parametrization style.

Author: stevenhua0320

tests/test_load_image.py

@@ -12,26 +12,26 @@
 # Hardcoded expected quantity
 # These values were obtained using the following scratch script:
 # arr = tiff.imread("docs/examples/KFe2As2-00838.tif")
-# arr.mean()
+# arr.mean(), arr.shape
 
 
 load_image_param = [
     # case 1: just filename of file in current directory.
     # expect function loads tiff file from cwd
-    ("KFe2As2-00838.tif", np.float32(2595.7087)),
+    "KFe2As2-00838.tif",
     # case 2: absolute file path to file in another directory.
     # expect file is found and correctly read.
-    ("home_dir/KFe2As2-00838.tif", np.float32(2595.7087)),
+    "home_dir/KFe2As2-00838.tif",
     # case 3: relative file path to file in another directory.
     # expect file is found and correctly read
-    ("./KFe2As2-00838.tif", np.float32(2595.7087)),
+    "./KFe2As2-00838.tif",
     # case 4: non-existent file that incurred by mistype.
-    ("nonexistent_file.tif", FileNotFoundError),
+    "nonexistent_file.tif",
 ]
 
 
-@pytest.mark.parametrize("file_name, expected", load_image_param)
-def test_load_image(file_name, expected, user_filesystem):
+@pytest.mark.parametrize("file_name", load_image_param)
+def test_load_image(file_name, user_filesystem):
     home_dir = user_filesystem["home"]
     cwd_dir = user_filesystem["cwd"]
     os.chdir(cwd_dir)
@@ -46,8 +46,11 @@ def test_load_image(file_name, expected, user_filesystem):
             "Cfg", (), {"fliphorizontal": False, "flipvertical": False}
         )()
         loader = LoadImage(cfg)
-        actual = loader.load_image(file_name).mean()
-        assert actual == expected
+        actual = loader.load_image(file_name)
+        expected_mean = np.float32(2595.7087)
+        expected_shape = (2048, 2048)
+        assert actual.shape == expected_shape
+        assert actual.mean() == expected_mean
     except FileNotFoundError:
         pytest.raises(
             FileNotFoundError,