update test

TPTBox/core/poi_fun/poi_global.py

@@ -31,6 +31,17 @@ def __init__(self, input_poi: poi.POI | POI_Descriptor):
             raise NotImplementedError(input_poi)
         self._centroids = global_points
 
+    def __str__(self) -> str:
+        return str(self._centroids)
+
+    @property
+    def zoom(self):
+        return (1, 1, 1)
+
+    @property
+    def origin(self):
+        return (0, 0, 0)
+
     @property
     def is_global(self) -> bool:
         """

unit_tests/test_centroids_save.py

@@ -37,7 +37,14 @@ def get_centroids(x: tuple[int, int, int] = (50, 30, 40), num_point=3, default_s
     for _ in range(num_point):
         point = tuple(random.randint(1, a * 100) / 100.0 for a in x)
         out_points[random.randint(1, 256), random.randint(1, 256) if default_sub is None else 50] = point
-    return POI(out_points, get_random_ax_code(), zoom=(random.random() * 3, random.random() * 3, random.random() * 3), shape=x)
+    return POI(
+        out_points,
+        get_random_ax_code(),
+        zoom=(random.random() * 3, random.random() * 3, random.random() * 3),
+        shape=x,
+        origin=(0, 0, 0),
+        rotation=np.eye(3),
+    )
 
 
 def get_centroids2(x: tuple[int, int, int] = (50, 30, 40), num_point=3):

unit_tests/test_poi_autogen.py

@@ -333,7 +333,7 @@ def test_remove_centroid(self):
         poi.centroids = {(5, 51): (1.0, 2.0, 3.0), (25, 12): (4.0, 5.0, 6.0)}
 
         # Remove a centroid using the remove_centroid method
-        poi.remove_centroid_((5, 51))
+        poi.remove_((5, 51))
 
         # Check that the centroid has been removed
         assert (5, 51) not in poi.centroids, poi
@@ -449,7 +449,7 @@ def test_remove_nonexistent_centroid(self):
         poi.centroids = {(12, 13): (1.0, 2.0, 3.0), (19, 23): (4.0, 5.0, 6.0)}
 
         # Remove a centroid that doesn't exist
-        poi.remove_centroid((50, 50))
+        poi.remove_((50, 50))
 
         # Check that the centroids are not modified
         assert poi.centroids == {(12, 13): (1.0, 2.0, 3.0), (19, 23): (4.0, 5.0, 6.0)}