use np.testing

Author: j-c-c

tests/_basis_util.py

@@ -208,7 +208,7 @@ def testExpand(self, basis):
         )
         assert isinstance(result, Coef)
         assert result.dtype == basis.coefficient_dtype
-        assert result.pixel_size == 1.234
+        np.testing.assert_array_equal(result.pixel_size, px_sz)
 
     def testInitWithIntSize(self, basis):
         # make sure we can instantiate with just an int as a shortcut

tests/test_class_src.py

@@ -209,7 +209,7 @@ class averages.
     assert test_src.symmetry_group == class_sim_fixture.symmetry_group
 
     # Check pixel_size pass-through.
-    assert test_src.pixel_size == class_sim_fixture.pixel_size
+    np.testing.assert_array_equal(test_src.pixel_size, class_sim_fixture.pixel_size)
 
 
 # Test the _HeapItem helper class

tests/test_covar2d_denoiser.py

@@ -113,7 +113,7 @@ def test_batched_rotcov2d_MSE(sim, basis):
     )
 
     # Test pixel_size pass-through.
-    assert sim.pixel_size == src.pixel_size
+    np.testing.assert_array_equal(sim.pixel_size, src.pixel_size)
 
 
 def test_source_mismatch(sim, basis):

tests/test_simulation.py

@@ -774,11 +774,11 @@ def test_pixel_size(caplog):
     vol = Volume(data, pixel_size=vol_px_sz)
 
     # Ensure vol pixel size
-    assert vol.pixel_size == vol_px_sz
+    np.testing.assert_array_equal(vol.pixel_size, vol_px_sz)
 
     # Generate Simulation and check pixel_size is inhereted from vol
     sim = Simulation(vols=vol)
-    assert sim.pixel_size == vol_px_sz
+    np.testing.assert_array_equal(sim.pixel_size, vol_px_sz)
 
     # Generate Simulation with provided pixel_size and check
     # that vol.pixel_size is overridden.
@@ -796,7 +796,7 @@ def test_pixel_size(caplog):
     sim = Simulation(vols=vol, pixel_size=sim_px_sz)
 
     assert msg in caplog.text
-    assert sim.pixel_size == sim_px_sz
+    np.testing.assert_array_equal(sim.pixel_size, sim_px_sz)
 
 
 def test_mismatched_pixel_size():