Add more evoked tests

cibr-jyu · Feb 28, 2024 · aafd72d · aafd72d
1 parent d380833
commit aafd72d
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 2 deletions.
diff --git a/meggie/actions/evoked_plot/tests/test_evoked_plot_all_channels.py b/meggie/actions/evoked_plot/tests/test_evoked_plot_all_channels.py
@@ -3,7 +3,7 @@
 from meggie.utilities.dialogs.outputOptionsMain import OutputOptions
 
 
-class TestEvokedPlot(BaseTestAction):
+class TestEvokedPlotAllChannels(BaseTestAction):
     def test_evoked_plot_all_channels(self):
 
         data = {"outputs": {"evoked": ["Evoked"]}}

diff --git a/meggie/actions/evoked_plot/tests/test_evoked_plot_channel_averages.py b/meggie/actions/evoked_plot/tests/test_evoked_plot_channel_averages.py
@@ -3,7 +3,7 @@
 from meggie.utilities.dialogs.outputOptionsMain import OutputOptions
 
 
-class TestEvokedPlot(BaseTestAction):
+class TestEvokedPlotChannelAverages(BaseTestAction):
     def test_evoked_plot_channel_averages(self):
 
         data = {"outputs": {"evoked": ["Evoked"]}}

diff --git a/meggie/actions/evoked_plot_topomap/tests/test_evoked_plot_topomap.py b/meggie/actions/evoked_plot_topomap/tests/test_evoked_plot_topomap.py
@@ -0,0 +1,22 @@
+from meggie.utilities.testing import BaseTestAction
+from meggie.actions.evoked_plot_topomap import PlotEvokedTopomap
+from meggie.actions.evoked_plot_topomap.dialogs.evokedTopomapDialogMain import (
+    EvokedTopomapDialog,
+)
+
+
+class TestEvokedPlotTopomap(BaseTestAction):
+    def test_evoked_plot_topomap(self):
+
+        data = {"outputs": {"evoked": ["Evoked"]}}
+
+        self.run_action(
+            action_name="evoked_plot_topomap",
+            handler=PlotEvokedTopomap,
+            data=data,
+            patch_paths=[
+                "meggie.actions.evoked_plot_topomap.dialogs.evokedTopomapDialogMain"
+            ],
+        )
+        dialog = self.find_dialog(EvokedTopomapDialog)
+        dialog.accept()
diff --git a/meggie/actions/evoked_save/tests/test_evoked_save_all_channels.py b/meggie/actions/evoked_save/tests/test_evoked_save_all_channels.py
@@ -0,0 +1,19 @@
+from meggie.utilities.testing import BaseTestAction
+from meggie.actions.evoked_save import SaveEvoked
+from meggie.utilities.dialogs.outputOptionsMain import OutputOptions
+
+
+class TestEvokedSaveAllChannels(BaseTestAction):
+    def test_evoked_save_all_channels(self):
+
+        data = {"outputs": {"evoked": ["Evoked"]}}
+
+        self.run_action(
+            action_name="evoked_save",
+            handler=SaveEvoked,
+            data=data,
+            patch_paths=["meggie.actions.evoked_save"],
+        )
+        dialog = self.find_dialog(OutputOptions)
+        dialog.ui.radioButtonChannelAverages.setChecked(False)
+        dialog.accept()
diff --git a/meggie/actions/evoked_save/tests/test_evoked_save_channel_averages.py b/meggie/actions/evoked_save/tests/test_evoked_save_channel_averages.py
@@ -0,0 +1,19 @@
+from meggie.utilities.testing import BaseTestAction
+from meggie.actions.evoked_save import SaveEvoked
+from meggie.utilities.dialogs.outputOptionsMain import OutputOptions
+
+
+class TestEvokedSaveChannelAverages(BaseTestAction):
+    def test_evoked_save_channel_averages(self):
+
+        data = {"outputs": {"evoked": ["Evoked"]}}
+
+        self.run_action(
+            action_name="evoked_save",
+            handler=SaveEvoked,
+            data=data,
+            patch_paths=["meggie.actions.evoked_save"],
+        )
+        dialog = self.find_dialog(OutputOptions)
+        dialog.ui.radioButtonChannelAverages.setChecked(True)
+        dialog.accept()