1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from unittest import TestCase
from typing import Tuple, Set
from AnyQt.QtCore import QItemSelectionModel
from AnyQt.QtGui import QStandardItemModel
from Orange.widgets.utils.itemselectionmodel import BlockSelectionModel, \
SymmetricSelectionModel
def selected(sel: QItemSelectionModel) -> Set[Tuple[int, int]]:
return set((r.row(), r.column()) for r in sel.selectedIndexes())
class TestBlockSelectionModel(TestCase):
def test_blockselectionmodel(self):
model = QStandardItemModel()
model.setRowCount(4)
model.setColumnCount(4)
sel = BlockSelectionModel(model)
sel.select(model.index(0, 0), BlockSelectionModel.Select)
self.assertSetEqual(selected(sel), {(0, 0)})
sel.select(model.index(0, 1), BlockSelectionModel.Select)
self.assertSetEqual(selected(sel), {(0, 0), (0, 1)})
sel.select(model.index(1, 1), BlockSelectionModel.Select)
self.assertSetEqual(selected(sel), {(0, 0), (0, 1), (1, 0), (1, 1)})
sel.select(model.index(0, 0), BlockSelectionModel.Deselect)
self.assertSetEqual(selected(sel), {(1, 1)})
sel.select(model.index(3, 3), BlockSelectionModel.ClearAndSelect)
self.assertSetEqual(selected(sel), {(3, 3)})
class TestSymmetricSelectionModel(TestCase):
def test_symmetricselectionmodel(self):
model = QStandardItemModel()
model.setRowCount(4)
model.setColumnCount(4)
sel = SymmetricSelectionModel(model)
sel.select(model.index(0, 0), BlockSelectionModel.Select)
self.assertSetEqual(selected(sel), {(0, 0)})
sel.select(model.index(0, 2), BlockSelectionModel.Select)
self.assertSetEqual(selected(sel), {(0, 0), (0, 2), (2, 0), (2, 2)})
sel.select(model.index(0, 0), BlockSelectionModel.Deselect)
self.assertSetEqual(selected(sel), {(2, 2)})
sel.select(model.index(2, 3), BlockSelectionModel.ClearAndSelect)
self.assertSetEqual(selected(sel), {(2, 2), (2, 3), (3, 2), (3, 3)})
self.assertSetEqual(set(sel.selectedItems()), {2, 3})
sel.setSelectedItems([1, 2])
self.assertSetEqual(set(sel.selectedItems()), {1, 2})
|
