[MRG] Added data check on axes in test_gui.py (#726)

* add data check on axes in `test_gui.py` Signed-off-by: samadpls <[email protected]> * refactored Signed-off-by: samadpls <[email protected]> * Added tests for GUI visualization features Signed-off-by: samadpls <[email protected]> * fixed linting error Signed-off-by: samadpls <[email protected]> * added doc string Signed-off-by: samadpls <[email protected]> * Remove unnecessary code assertions Signed-off-by: samadpls <[email protected]> * Refactor visualization test Signed-off-by: samadpls <[email protected]> * refactor --------- Signed-off-by: samadpls <[email protected]>
jonescompneurolab · Mar 13, 2024 · 21dcea5 · 21dcea5
1 parent 3c205e5
commit 21dcea5
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/hnn_core/tests/test_gui.py b/hnn_core/tests/test_gui.py
@@ -396,6 +396,7 @@ def test_gui_figure_overlay():
 
 
 def test_gui_adaptive_spectrogram():
+    """Test the adaptive spectrogram functionality of the HNNGUI."""
     gui = HNNGUI()
     gui.compose()
     gui.params['N_pyr_x'] = 3
@@ -422,6 +423,34 @@ def test_gui_adaptive_spectrogram():
     plt.close('all')
 
 
+@pytest.fixture
+def setup_gui():
+    gui = HNNGUI()
+    gui.compose()
+    gui.params['N_pyr_x'] = 3
+    gui.params['N_pyr_y'] = 3
+    gui.run_button.click()
+    return gui
+
+
+@pytest.mark.parametrize("viz_type", ["layer2 dipole", "layer5 dipole",
+                                      "spikes", "PSD", "network"])
+def test_gui_visualization(setup_gui, viz_type):
+    """Test visualization functionality in the HNNGUI."""
+    gui = setup_gui
+    figid = 1
+    figname = f'Figure {figid}'
+    axname = 'ax1'
+    gui._simulate_viz_action("edit_figure", figname,
+                             axname, 'default', viz_type, {}, 'clear')
+    gui._simulate_viz_action("edit_figure", figname,
+                             axname, 'default', viz_type, {}, 'plot')
+    # Check if data is plotted on the axes
+    assert len(gui.viz_manager.figs[figid].axes) == 2
+    # Check default figs have data on their axis
+    assert gui.viz_manager.figs[figid].axes[1].has_data()
+
+
 def test_unlink_relink_widget():
     """Tests the unlinking and relinking of widgets decorator."""