ListViewModel: Add tests

Author: janezd

Orange/widgets/tests/test_gui.py

@@ -1,9 +1,11 @@
+from Orange.data import ContinuousVariable
 from Orange.widgets import gui
 from Orange.widgets.tests.base import GuiTest
+from Orange.widgets.utils.itemmodels import VariableListModel
 from Orange.widgets.widget import OWWidget
 
 
-class OWTestDoubleSpin(GuiTest):
+class TestDoubleSpin(GuiTest):
     some_param = 0
     some_option = False
 
@@ -13,3 +15,23 @@ def test_checked_extension(self):
         widget = OWWidget()
         gui.doubleSpin(widget=widget, master=self, value="some_param",
                        minv=1, maxv=10, checked="some_option")
+
+
+class TestListModel(GuiTest):
+    def test_select(self):
+        widget = OWWidget()
+        widget.foo = None
+        self.attrs = VariableListModel()
+        view = gui.listView(widget.controlArea, widget, "foo", model=self.attrs)
+        self.assertIsNone(widget.foo)
+        a, b, c = (ContinuousVariable(x) for x in "abc")
+        self.attrs[:] = [a, b, c]
+        view.setCurrentIndex(self.attrs.index(0, 0))
+        self.assertIs(widget.foo, a)
+        view.setCurrentIndex(self.attrs.index(2, 0))
+        self.assertIs(widget.foo, c)
+
+        widget.foo = b
+        selection = view.selectedIndexes()
+        self.assertEqual(len(selection), 1)
+        self.assertEqual(selection[0].row(), 1)