Test and Score: Add tests

Author: janezd

Orange/widgets/evaluate/tests/test_owtestlearners.py

@@ -1,8 +1,14 @@
 # pylint: disable=missing-docstring
+from unittest.mock import patch
+
+import collections
 import numpy as np
 
 import unittest
 
+from AnyQt.QtWidgets import QMenu
+from AnyQt.QtCore import QPoint
+
 from Orange.data import Table, Domain
 from Orange.classification import MajorityLearner
 from Orange.regression import MeanLearner
@@ -70,6 +76,59 @@ def test_CrossValidationByFeature(self):
         self.assertEqual(self.widget.resampling, OWTestLearners.KFold)
         self.assertFalse(self.widget.features_combo.isEnabled())
 
+    def test_update_shown_columns(self):
+        w = self.widget  #: OWTestLearners
+        all, shown = "MABDEFG", "ABDF"
+        header = w.view.horizontalHeader()
+        w.shown_scores = set(shown)
+        w.result_model.setHorizontalHeaderLabels(list(all))
+        w._update_shown_columns()
+        for i, name in enumerate(all):
+            self.assertEqual(name == "M" or name in shown,
+                             not header.isSectionHidden(i),
+                             msg="error in section {}({})".format(i, name))
+
+        w.shown_scores = set()
+        w._update_shown_columns()
+        for i, name in enumerate(all):
+            self.assertEqual(i == 0,
+                             not header.isSectionHidden(i),
+                             msg="error in section {}({})".format(i, name))
+
+    def test_show_column_chooser(self):
+        w = self.widget  #: OWTestLearners
+        all, shown = "MABDEFG", "ABDF"
+        header = w.view.horizontalHeader()
+        w.shown_scores = set(shown)
+        w.result_model.setHorizontalHeaderLabels(list(all))
+        w._update_shown_columns()
+
+        actions = collections.OrderedDict()
+        menu_add_action = QMenu.addAction
+
+        def addAction(menu, a):
+            action = menu_add_action(menu, a)
+            actions[a] = action
+            return action
+
+        def execmenu(*_):
+            self.assertEqual(list(actions), list(all)[1:])
+            print(actions)
+            for name, action in actions.items():
+                self.assertEqual(action.isChecked(), name in shown)
+            actions["E"].triggered.emit(True)
+            self.assertEqual(w.shown_scores, set("ABDEF"))
+            actions["B"].triggered.emit(False)
+            self.assertEqual(w.shown_scores, set("ADEF"))
+            for i, name in enumerate(all):
+                self.assertEqual(name == "M" or name in "ADEF",
+                                 not header.isSectionHidden(i),
+                                 msg="error in section {}({})".format(i, name))
+
+        with patch("AnyQt.QtWidgets.QMenu.addAction", addAction),\
+                patch("AnyQt.QtWidgets.QMenu.exec", execmenu):
+            w.show_column_chooser(QPoint(0, 0))
+
 
 class TestHelpers(unittest.TestCase):
     def test_results_one_vs_rest(self):