tests: basic gui test

Author: paulmueller

shapeout2/gui/main.py

@@ -278,13 +278,16 @@ def adopt_slot(self, slot_state):
             raise ValueError("Slot not in pipeline: {}".format(slot_id))
         self.adopt_pipeline(state)
 
-    def add_dataslot(self):
+    def add_dataslot(self, paths=None):
         """Adds a dataslot to the pipeline"""
-        fnames, _ = QtWidgets.QFileDialog.getOpenFileNames(
-            parent=self,
-            caption="Select an RT-DC measurement",
-            directory=self.settings.get_path(name="rtdc import dataset"),
-            filter="RT-DC Files (*.rtdc)")
+        if paths is None:
+            fnames, _ = QtWidgets.QFileDialog.getOpenFileNames(
+                parent=self,
+                caption="Select an RT-DC measurement",
+                directory=self.settings.get_path(name="rtdc import dataset"),
+                filter="RT-DC Files (*.rtdc)")
+        else:
+            fnames = paths
 
         if fnames:
             self.toolButton_new_plot.setEnabled(True)

shapeout2/gui/matrix/block_matrix.py

@@ -50,6 +50,45 @@ def __setstate__(self, state):
     def adopt_pipeline(self, pipeline_state):
         self.__setstate__(pipeline_state)
 
+    def get_widget(self, slot_id=None, filt_plot_id=None):
+        """Convenience function for testing"""
+        if slot_id is None and filt_plot_id is not None:
+            # get a filter or a plot
+            w = self.data_matrix.filter_widgets + self.plot_matrix.plot_widgets
+            for wi in w:
+                if wi.identifier == filt_plot_id:
+                    break
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        elif slot_id is not None and filt_plot_id is None:
+            # get a slot
+            for wi in self.data_matrix.dataset_widgets:
+                if wi.identifier == slot_id:
+                    break
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        elif slot_id is not None and filt_plot_id is not None:
+            # get a matrix element
+            wd = self.data_matrix.element_widget_dict
+            wp = self.plot_matrix.element_widget_dict
+            fpd = wd[slot_id]
+            fpp = wp[slot_id]
+            if filt_plot_id in fpp:
+                wi = fpp[filt_plot_id]
+            elif filt_plot_id in fpd:
+                wi = fpd[filt_plot_id]
+            else:
+                raise KeyError(
+                    "Widget identifier '{}' not found!".format(filt_plot_id))
+            return wi
+        else:
+            raise ValueError(
+                "One of `slot_id` or `filt_plot_id` must be specified!")
+
     def invalidate_elements(self, invalid_dm, invalid_pm):
         for slot_id, filt_id in invalid_dm:
             em = self.data_matrix.get_matrix_element(slot_id, filt_id)

shapeout2/gui/matrix/data_matrix.py

@@ -158,6 +158,17 @@ def element_width(self):
             width = 67
         return width
 
+    @property
+    def element_widget_dict(self):
+        els = {}
+        for ii, ws in enumerate(self.dataset_widgets):
+            elsd = {}
+            for jj, wf in enumerate(self.filter_widgets):
+                it = self.glo.itemAtPosition(ii+1, jj+1)
+                elsd[wf.identifier] = it.widget()
+            els[ws.identifier] = elsd
+        return els
+
     @property
     def filter_widgets(self):
         filters = []

shapeout2/gui/matrix/plot_matrix.py

@@ -90,6 +90,17 @@ def data_matrix(self):
             raise KeyError("DataMatrix not found!")
         return ch
 
+    @property
+    def element_widget_dict(self):
+        els = {}
+        for ii, ws in enumerate(self.data_matrix.dataset_widgets):
+            elsd = {}
+            for jj, wf in enumerate(self.plot_widgets):
+                it = self.glo.itemAtPosition(ii+1, jj)
+                elsd[wf.identifier] = it.widget()
+            els[ws.identifier] = elsd
+        return els
+
     @property
     def element_width(self):
         """Data matrix element width (without 2px spacing)"""

tests/test_gui_basic.py

@@ -0,0 +1,41 @@
+"""Test of data set functionalities"""
+import pathlib
+
+from PyQt5 import QtCore
+
+from shapeout2.gui.main import ShapeOut2
+
+
+def test_simple(qtbot):
+    """Open the main window and close it again"""
+    main_window = ShapeOut2()
+    main_window.close()
+
+
+def test_matrix_slots(qtbot):
+    mw = ShapeOut2()
+    qtbot.addWidget(mw)
+
+    # add a dataslot
+    path = pathlib.Path(__file__).parent / "data" / "calibration_beads_47.rtdc"
+    mw.add_dataslot(paths=[path])
+    # add another one
+    mw.add_dataslot(paths=[path])
+
+    assert len(mw.pipeline.slot_ids) == 2, "we added those"
+    assert len(mw.pipeline.filter_ids) == 1, "automatically added"
+
+    # activate a dataslot
+    slot_id = mw.pipeline.slot_ids[0]
+    filt_id = mw.pipeline.filter_ids[0]
+    em = mw.block_matrix.get_widget(slot_id, filt_id)
+    qtbot.mouseClick(em, QtCore.Qt.LeftButton)
+    # did that work?
+    assert mw.pipeline.is_element_active(slot_id, filt_id)
+    slot_id2 = mw.pipeline.slot_ids[1]
+    assert not mw.pipeline.is_element_active(slot_id2, filt_id)
+
+    # remove a dataslot
+    wd = mw.block_matrix.get_widget(slot_id=slot_id)
+    wd.action_remove()
+    assert not mw.pipeline.is_element_active(slot_id2, filt_id)