Ensure example_mouse_100um is loaded for testing the registration_widget

brainglobe · Dec 20, 2024 · b073fed · b073fed
1 parent f53e02c
commit b073fed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -118,6 +118,7 @@ def setup_preexisting_local_atlases():
     in the test user data."""
     preexisting_atlases = [
         ("example_mouse_100um", "v1.2"),
+        ("allen_mouse_25um", "v1.2"),
         ("allen_mouse_100um", "v1.2"),
         ("osten_mouse_100um", "v1.1"),
     ]

diff --git a/tests/test_registration_widget.py b/tests/test_registration_widget.py
@@ -29,7 +29,12 @@ def registration_widget_with_example_atlas(make_napari_viewer_with_images):
 
     # Based on the downloaded atlases by the fixture in conftest.py,
     # the example atlas will be in the third position.
-    example_atlas_index = 2
+    example_atlas_index = -1
+    for i, atlas in enumerate(widget._available_atlases):
+        if atlas == "example_mouse_100um":
+            example_atlas_index = i
+            break
+
     widget._on_atlas_dropdown_index_changed(example_atlas_index)
 
     return widget