tests: Use pytest.approx() in float comparisons

Tests were failing, hopefully this fixes that by using `pytest.approx()`.

Author: ns-rse

tests/test_gwy.py

@@ -19,9 +19,9 @@ def test_load_gwy() -> None:
     result_image, result_pixel_to_nm_scaling = gwy.load_gwy(file_path, channel=channel)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == (512, 512)
-    assert result_image.sum() == 33836850.232917726
+    assert result_image.sum() == pytest.approx(33836850.232917726)
     assert isinstance(result_pixel_to_nm_scaling, float)
-    assert result_pixel_to_nm_scaling == 0.8468632812499975
+    assert result_pixel_to_nm_scaling == pytest.approx(0.8468632812499975)
 
 
 def test_gwy_read_object() -> None:

tests/test_ibw.py

@@ -30,11 +30,11 @@ def test_load_ibw(
     file_path = RESOURCES / file_name
     result_image, result_pixel_to_nm_scaling = ibw.load_ibw(file_path, channel)  # type: ignore
 
-    assert result_pixel_to_nm_scaling == pixel_to_nm_scaling
+    assert result_pixel_to_nm_scaling == pytest.approx(pixel_to_nm_scaling)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == image_shape
     assert result_image.dtype == image_dtype
-    assert result_image.sum() == image_sum
+    assert result_image.sum() == pytest.approx(image_sum)
 
 
 def test_load_ibw_file_not_found() -> None:

tests/test_jpk.py

@@ -78,11 +78,11 @@ def test_load_jpk(
     file_path = RESOURCES / file_name
     result_image, result_pixel_to_nm_scaling = jpk.load_jpk(file_path, channel)  # type: ignore
 
-    assert result_pixel_to_nm_scaling == pixel_to_nm_scaling
+    assert result_pixel_to_nm_scaling == pytest.approx(pixel_to_nm_scaling)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == image_shape
     assert result_image.dtype == image_dtype
-    assert result_image.sum() == image_sum
+    assert result_image.sum() == pytest.approx(image_sum)
 
 
 def test_load_jpk_file_not_found() -> None:

tests/test_spm.py

@@ -39,11 +39,11 @@ def test_load_spm(
     file_path = RESOURCES / file_name
     result_image, result_pixel_to_nm_scaling = spm.load_spm(file_path, channel=channel)
 
-    assert result_pixel_to_nm_scaling == pixel_to_nm_scaling
+    assert result_pixel_to_nm_scaling == pytest.approx(pixel_to_nm_scaling)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == image_shape
     assert result_image.dtype == image_dtype
-    assert result_image.sum() == image_sum
+    assert result_image.sum() == pytest.approx(image_sum)
 
 
 @patch("pySPM.SPM.SPM_image.pxs")

tests/test_stp.py

@@ -32,8 +32,8 @@ def test_load_stp(
     file_path = RESOURCES / file_name
     result_image, result_pixel_to_nm_scaling = load_stp(file_path=file_path)
 
-    assert result_pixel_to_nm_scaling == expected_pixel_to_nm_scaling
+    assert result_pixel_to_nm_scaling == pytest.approx(expected_pixel_to_nm_scaling)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == expected_image_shape
     assert result_image.dtype == expected_image_dtype
-    assert result_image.sum() == expected_image_sum
+    assert result_image.sum() == pytest.approx(expected_image_sum)

tests/test_top.py

@@ -32,8 +32,8 @@ def test_load_top(
     file_path = RESOURCES / file_name
     result_image, result_pixel_to_nm_scaling = load_top(file_path=file_path)
 
-    assert result_pixel_to_nm_scaling == expected_pixel_to_nm_scaling
+    assert result_pixel_to_nm_scaling == pytest.approx(expected_pixel_to_nm_scaling)
     assert isinstance(result_image, np.ndarray)
     assert result_image.shape == expected_image_shape
     assert result_image.dtype == expected_image_dtype
-    assert result_image.sum() == expected_image_sum
+    assert result_image.sum() == pytest.approx(expected_image_sum)

tests/test_topostats.py

@@ -96,9 +96,9 @@ def test_load_topostats(
         assert topostats_data[version_key] == float(version)
     else:
         assert topostats_data[version_key] == version
-    assert topostats_data["pixel_to_nm_scaling"] == pixel_to_nm_scaling
+    assert topostats_data["pixel_to_nm_scaling"] == pytest.approx(pixel_to_nm_scaling)
     assert topostats_data["image"].shape == image_shape
-    assert topostats_data["image"].sum() == image_sum
+    assert topostats_data["image"].sum() == pytest.approx(image_sum)
     if version >= "0.2":
         assert isinstance(topostats_data["img_path"], Path)