Changed test for invert method

usnistgov · Jul 16, 2024 · 68ede97 · 68ede97
1 parent 3b2c2f3
commit 68ede97
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tomotools/tests/test_base.py b/tomotools/tests/test_base.py
@@ -86,8 +86,9 @@ def test_stack_normalize(self):
         assert norm.data.min() == 0.0
 
     def test_stack_invert(self):
-        # stack = ds.get_needle_data()
-        stack = CommonStack(np.random.random([10, 100, 100]))
+	im = np.zeros([10,100,100])
+	im[:,40:60,40:60] = 10
+        stack = CommonStack(im)
         invert = stack.invert()
         hist, bins = np.histogram(stack.data)
         hist_inv, bins_inv = np.histogram(invert.data)