update tests

brainglobe_utils/brainmapper/transform_widget.py

@@ -352,7 +352,8 @@ def run_transform_downsampled_points_to_atlas_space(self):
             display_info(
                 self,
                 "Points outside atlas",
-                f"{len(points_out_of_bounds)} fell outside the atlas space",
+                f"{len(points_out_of_bounds)} "
+                f"points fell outside the atlas space",
             )
 
     def get_downsampled_space(self):

tests/tests/test_brainmapper/test_transform_widget.py

@@ -43,6 +43,14 @@
     ]
 )
 
+points_outside_brain = np.array(
+    [
+        [10000, 10000, 10000],
+        [100001, 100001, 100001],
+    ]
+)
+
+
 points_in_downsampled_space = np.array(
     [
         [10.0, 68.0, 105.0],
@@ -264,6 +272,22 @@ def test_transform_points_to_atlas_space(
     )
 
 
+def test_transformation_raises_info_points_out_of_bounds(
+    transformation_widget_with_data, mock_display_info
+):
+    points_layer = transformation_widget_with_data.viewer.add_points(
+        points_outside_brain
+    )
+    transformation_widget_with_data.points_layer = points_layer
+    transformation_widget_with_data.run_transform_points_to_downsampled_space()
+    transformation_widget_with_data.run_transform_downsampled_points_to_atlas_space()
+    mock_display_info.assert_called_once_with(
+        transformation_widget_with_data,
+        "Points outside atlas",
+        "2 points fell outside the atlas space",
+    )
+
+
 def test_check_layers(transformation_widget_with_data):
     assert transformation_widget_with_data.check_layers()