TST: itemmodels.AbstractSortTableModel: add test case, test sorting

Author: kernc

Orange/widgets/tests/test_itemmodels.py

@@ -7,7 +7,7 @@
 
 from Orange.data import Domain, ContinuousVariable
 from Orange.widgets.utils.itemmodels import \
-    PyTableModel, PyListModel, DomainModel, _argsort
+    AbstractSortTableModel, PyTableModel, PyListModel, DomainModel, _argsort
 
 
 class TestArgsort(TestCase):
@@ -120,6 +120,20 @@ def test_other_roles(self):
                                         Qt.TextAlignmentRole))
 
 
+class TestAbstractSortTableModel(TestCase):
+    def setUp(self):
+        assert issubclass(PyTableModel, AbstractSortTableModel)
+        self.model = PyTableModel([[1, 4],
+                                   [2, 3]])
+
+    def test_sorting(self):
+        self.model.sort(1, Qt.AscendingOrder)
+        self.assertSequenceEqual(self.model.mapToSourceRows(...).tolist(), [1, 0])
+
+        self.model.sort(1, Qt.DescendingOrder)
+        self.assertSequenceEqual(self.model.mapToSourceRows(...).tolist(), [0, 1])
+
+
 class TestPyListModel(TestCase):
     @classmethod
     def setUpClass(cls):