test: updated test with explicit title checks

jonescompneurolab · Sep 23, 2024 · 13b1a52 · 13b1a52
1 parent 0ee4eb6
commit 13b1a52
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/hnn_core/tests/test_gui.py b/hnn_core/tests/test_gui.py
@@ -1115,6 +1115,17 @@ def test_delete_single_drive(setup_gui):
     """Deleting a single drive."""
     gui = setup_gui
     assert len(gui.drive_accordion.children) == 6
-
-    gui._simulate_delete_single_drive(0)
+    assert gui.drive_accordion.titles == ('evdist1 (distal)',
+                                          'evprox1 (proximal)',
+                                          'evprox2 (proximal)',
+                                          'alpha_prox (proximal)',
+                                          'poisson (proximal)',
+                                          'tonic')
+
+    gui._simulate_delete_single_drive(2)
     assert len(gui.drive_accordion.children) == 5
+    assert gui.drive_accordion.titles == ('evdist1 (distal)',
+                                          'evprox1 (proximal)',
+                                          'alpha_prox (proximal)',
+                                          'poisson (proximal)',
+                                          'tonic')