test_OWGeneSets: update tests

Author: JakaKokosar

orangecontrib/bioinformatics/tests/widgets/test_OWGeneSets.py

@@ -1,11 +1,13 @@
 import os
-from unittest.mock import Mock
+
+from AnyQt.QtCore import Qt, QItemSelectionModel
 
 from orangewidget.tests.base import WidgetTest
 
 from Orange.data import Table, Domain, StringVariable
 
-from orangecontrib.bioinformatics.widgets.OWGeneSets import OWGeneSets
+from orangecontrib.bioinformatics.geneset import GeneSets
+from orangecontrib.bioinformatics.widgets.OWGeneSets import OWGeneSets, GeneSetsModel
 from orangecontrib.bioinformatics.widgets.utils.data import TableAnnotation
 
 
@@ -15,33 +17,49 @@ def setUp(self) -> None:
         self.widget = self.create_widget(OWGeneSets)
 
         domain = Domain([], metas=[StringVariable('name'), StringVariable('Entrez ID')])
-        self.data = Table.from_list(domain, [['CA1', '759'], ['CA2', '760'], ['CA3', '761']])
+        self.data = Table.from_list(
+            domain, [['CA1', '759'], ['CA2', '760'], ['CA3', '761']]
+        )
         self.data.attributes[TableAnnotation.tax_id] = '9606'
         self.data.attributes[TableAnnotation.gene_as_attr_name] = False
         self.data.attributes[TableAnnotation.gene_id_column] = 'Entrez ID'
 
-    def test_input_info(self):
-        input_sum = self.widget.info.set_input_summary = Mock()
-
-        self.send_signal(self.widget.Inputs.data, self.data)
+    def test_gene_sets_loaded(self):
         self.wait_until_finished()
-        input_sum.assert_called_with("3", "3 unique gene names on input.\n")
+        self.assertTrue(self.widget.filter_proxy_model.sourceModel().rowCount())
+        self.assertTrue(self.widget.filter_proxy_model.sourceModel().columnCount())
 
-        self.send_signal(self.widget.Inputs.custom_gene_sets, self.data)
-        self.wait_until_finished()
-        input_sum.assert_called_with("3|3", "3 unique gene names on input.\n3 marker genes in 3 sets\n")
+    def test_gene_set_selection(self):
+        first_row = self.widget.view.model().index(0, 0)
+        second_row = self.widget.view.model().index(1, 0)
 
-        self.send_signal(self.widget.Inputs.data, None)
-        self.send_signal(self.widget.Inputs.custom_gene_sets, None)
-        self.wait_until_finished()
-        input_sum.assert_called_with(self.widget.info.NoInput)
+        selection_model = self.widget.view.selectionModel()
+        selection_flags = QItemSelectionModel.Select | QItemSelectionModel.Rows
+        selection_model.select(first_row, selection_flags)
+        selection_model.select(second_row, selection_flags)
 
-    def test_output_info(self):
-        output_sum = self.widget.info.set_output_summary = Mock()
+        selection = selection_model.selectedRows(column=GeneSetsModel.genes_in_set)
+        self.assertTrue(len(selection) == 2)
 
+        output = self.get_output(self.widget.Outputs.gene_sets)
+        self.assertIsInstance(output, GeneSets)
+        self.assertTrue(len(output) == 2)
+
+    def test_input_data(self):
         self.send_signal(self.widget.Inputs.data, self.data)
         self.wait_until_finished()
-        output_sum.assert_called_with("0", "0 genes on output.\n")
 
-        self.send_signal(self.widget.Inputs.data, None)
-        output_sum.assert_called_with(self.widget.info.NoOutput)
+        first_row = self.widget.view.model().index(0, 0)
+        tenth_row = self.widget.view.model().index(10, 0)
+
+        selection_model = self.widget.view.selectionModel()
+        selection_flags = QItemSelectionModel.Select | QItemSelectionModel.Rows
+        selection_model.select(first_row, selection_flags)
+        selection_model.select(tenth_row, selection_flags)
+
+        selection = selection_model.selectedRows(column=GeneSetsModel.mapped_genes)
+
+        # all genes from the data are found in first gene set
+        self.assertTrue(selection[0].data(Qt.DisplayRole))
+        # non genes map to 10th gene set
+        self.assertFalse(selection[1].data(Qt.DisplayRole))