tests updated

teamtomo · Sep 23, 2024 · 2f2968b · 2f2968b
1 parent 9c39a50
commit 2f2968b
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/test_functions.py b/tests/test_functions.py
@@ -2,20 +2,20 @@
 import numpy as np
 
 def test_cone():
-    mask = cone(100, 50, 50, 0, 1)
+    mask = cone(100, 50, 50, 0, 1, "standard")
     assert mask.shape == (100, 100, 100)
     assert mask.sum() > np.pi * 24**2 * (50 / 3) # Volume of cone
     assert mask.sum() < np.pi * 25**2 * 50 # Volume of cylinder
 
 def test_cube():
-    mask = cube(100, 50, 0, 1, 0)
+    mask = cube(100, 50, 0, 1, 0, "standard")
     assert mask.shape == (100, 100, 100)
     # Test against volume of cube +- center and subpixel issues
     assert mask.sum() > 48**3
     assert mask.sum() < 52**3
 
 def test_cuboid():
-    mask = cuboid(100, (50,40,30), 0, 1, 0)
+    mask = cuboid(100, (50,40,30), 0, 1, 0, "standard")
     assert mask.shape == (100, 100, 100)
     # Test against volume of cuboid +- center and subpixel issues
     assert mask.sum() > 48 * 38 * 28
@@ -28,26 +28,26 @@ def test_cuboid():
 #    assert mask.sum() < 2 * np.pi * 25 * 50 # Area of cylinder
 
 def test_cylinder():
-    mask = cylinder(100, 50, 50, 0, 0, 1)
+    mask = cylinder(100, 50, 50, 0, 0, 1, "standard")
     assert mask.shape == (100, 100, 100)
     assert mask.sum() > np.pi * 25**2 * 48 # Volume of cylinder
     assert mask.sum() < np.pi * 25**2 * 51 # Volume of cylinder
 
 def test_ellipsoid():
-    mask = ellipsoid(100, (50,40,30), 0, 1,0)
+    mask = ellipsoid(100, (50,40,30), 0, 1,0, "standard")
     assert mask.shape == (100, 100, 100)
     # Test against volume of ellipsoid +- center and subpixel issues
     assert mask.sum() > 24 * 19 * 14 * 4/3 * np.pi
     assert mask.sum() < 26 * 21 * 16 * 4/3 * np.pi
 
 def test_sphere():
-    mask = sphere(100, 50, 0, 1,0)
+    mask = sphere(100, 50, 0, 1,0, "standard")
     assert mask.shape == (100, 100, 100)
     assert mask.sum() > 4/3 * np.pi * 24**3 # Volume of sphere
     assert mask.sum() < 4/3 * np.pi * 26**3 # Volume of sphere
 
 def test_tube():
-    mask = tube(100, 50, 50, 0, 0, 1)
+    mask = tube(100, 50, 50, 0, 0, 1, "standard")
     assert mask.shape == (100, 100, 100)
     assert mask.sum() > np.pi * 24**2 * 48 # Volume of tube
     assert mask.sum() < np.pi * 26**2 * 52 # Volume of tube