Updated tests for xray view

PMEAL · Apr 9, 2024 · d5b8e17 · d5b8e17
1 parent b8972bf
commit d5b8e17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/unit/test_visualization.py b/test/unit/test_visualization.py
@@ -23,15 +23,15 @@ def test_sem_y(self):
 
     def test_xray_y(self):
         xray = ps.visualization.xray(self.im, axis=1)
-        assert np.sum(xray) == np.sum(~self.im)
+        assert np.min(xray) >= 0 and np.max(xray) <= 1
 
     def test_sem_z(self):
         sem = ps.visualization.sem(self.im, axis=2)
         assert sem.ndim == 2
 
     def test_xray_z(self):
         xray = ps.visualization.xray(self.im, axis=2)
-        assert np.sum(xray) == np.sum(~self.im)
+        assert np.min(xray) >= 0 and np.max(xray) <= 1
 
     def test_imshow_single(self):
         im = ps.generators.blobs(shape=[10, 20, 30])