def setup_scene(self, parts, data):
# type: (Parts, Table) -> None
widget = HeatmapGridWidget()
widget.setColorMap(self.color_map())
self.scene.addItem(widget)
self.scene.widget = widget
columns = [v.name for v in data.domain.attributes]
parts = HeatmapGridWidget.Parts(
rows=[
HeatmapGridWidget.RowItem(r.title, r.indices, r.cluster)
for r in parts.rows
],
columns=[
HeatmapGridWidget.ColumnItem(c.title, c.indices, c.cluster)
for c in parts.columns
],
data=data.X,
span=parts.span,
row_names=None,
col_names=columns,
)
widget.setHeatmaps(parts)
side = self.row_side_colors()
if side is not None:
widget.setRowSideColorAnnotations(side[0],
side[1],
name=side[2].name)
widget.setColumnLabelsPosition(self._column_label_pos)
widget.setAspectRatioMode(
Qt.KeepAspectRatio if self.keep_aspect else Qt.IgnoreAspectRatio)
widget.setShowAverages(self.averages)
widget.setLegendVisible(self.legend)
widget.layoutDidActivate.connect(self.__on_layout_activate)
widget.selectionFinished.connect(self.on_selection_finished)
self.update_annotations()
self.view.setCentralWidget(widget)
self.parts = parts
class TestHeatmapGridWidget(GuiTest):
scene: QGraphicsScene
view: QGraphicsView
def setUp(self) -> None:
super().setUp()
self.view = QGraphicsView()
self.scene = QGraphicsScene(self.view)
self.view.setScene(self.scene)
def tearDown(self) -> None:
self.scene.clear()
self.scene.deleteLater()
self.scene = None
self.view.deleteLater()
self.view = None
super().tearDown()
_c2 = Tree(ClusterData((0, 1), 0.5), (
Tree(SingletonData((0, 0), 0, 0), ()),
Tree(SingletonData((1, 1), 0, 1), ()),
))
_Data = {
"0-0":
HeatmapGridWidget.Parts(rows=[],
columns=[],
data=np.zeros(shape=(0, 0)),
span=(0, 0)),
"1-0":
HeatmapGridWidget.Parts(rows=[],
columns=[],
data=np.zeros(shape=(0, 0)),
span=(0, 0)),
"0-1":
HeatmapGridWidget.Parts(
rows=[],
columns=[HeatmapGridWidget.ColumnItem("a", [0])],
data=np.zeros(shape=(0, 1)),
span=(0, 1)),
"1-1":
HeatmapGridWidget.Parts(
rows=[HeatmapGridWidget.RowItem("a", [0])],
columns=[HeatmapGridWidget.ColumnItem("a", [0])],
data=np.zeros(shape=(1, 1)),
span=(0, 1),
row_names=["a"],
col_names=["b"],
),
"2-2-split":
HeatmapGridWidget.Parts(
rows=[
HeatmapGridWidget.RowItem("a", [0]),
HeatmapGridWidget.RowItem("b", [1]),
],
columns=[
HeatmapGridWidget.ColumnItem("a", [0]),
HeatmapGridWidget.ColumnItem("a", [1]),
],
data=np.zeros(shape=(2, 2)),
span=(-1, 1),
row_names=["a", "b"],
col_names=["b", "b"],
),
"2-2-cl":
HeatmapGridWidget.Parts(
rows=[HeatmapGridWidget.RowItem("", [0, 1], _c2)],
columns=[HeatmapGridWidget.ColumnItem("", [0, 1], _c2)],
data=np.zeros(shape=(2, 2)),
span=(-1, 1),
row_names=["a", "b"],
col_names=["b", "b"],
),
"2-2":
HeatmapGridWidget.Parts(
rows=[HeatmapGridWidget.RowItem("", [0, 1])],
columns=[HeatmapGridWidget.ColumnItem("", [0, 1])],
data=np.zeros(shape=(2, 2)),
span=(-1, 1),
row_names=["a", "b"],
col_names=["b", "b"],
)
}
def test_widget(self):
w = HeatmapGridWidget()
self.scene.addItem(w)
for p in self._Data.values():
w.setHeatmaps(p)
w.headerGeometry()
w.footerGeometry()
def test_widget_annotations(self):
w = HeatmapGridWidget()
self.scene.addItem(w)
w.setHeatmaps(self._Data["2-2"])
# Coverage. The game.
w.setLegendVisible(True)
w.setLegendVisible(False)
w.setShowAverages(True)
w.setShowAverages(False)
w.setRowLabels(None)
w.setRowLabels(["1", "2"])
w.setRowLabelsVisible(False)
w.setRowLabelsVisible(True)
w.setColumnLabels(None)
w.setColumnLabels(["1", "2"])
w.setAspectRatioMode(Qt.IgnoreAspectRatio)
w.setAspectRatioMode(Qt.KeepAspectRatio)
w.setAspectRatioMode(Qt.KeepAspectRatioByExpanding)
for pos in (
HeatmapGridWidget.NoPosition,
HeatmapGridWidget.PositionTop,
HeatmapGridWidget.PositionBottom,
HeatmapGridWidget.PositionTop
| HeatmapGridWidget.PositionBottom,
):
w.setColumnLabelsPosition(pos)
w.setRowSideColorAnnotations(
np.array([0, 1]),
CategoricalColorMap(np.array([[255] * 3, [0] * 3]),
names=["a", "b"]))
w.setRowSideColorAnnotations(None)
def test_selection(self):
w = HeatmapGridWidget()
self.scene.addItem(w)
w.setHeatmaps(self._Data["2-2"])
view = self.view
w.resize(w.effectiveSizeHint(Qt.PreferredSize))
h = w.layout().itemAt(w.Row0, w.Col0 + 1)
pos = view.mapFromScene(h.scenePos())
spy = QSignalSpy(w.selectionFinished)
QTest.mouseClick(view.viewport(),
Qt.LeftButton,
pos=pos + QPoint(1, 1))
self.assertSequenceEqual(list(spy), [[]])
self.assertSequenceEqual(w.selectedRows(), [0])
spy = QSignalSpy(w.selectionFinished)
QTest.mouseClick(view.viewport(),
Qt.LeftButton,
Qt.ControlModifier,
pos=pos + QPoint(1, 1))
self.assertSequenceEqual(list(spy), [[]])
self.assertSequenceEqual(w.selectedRows(), [])
spy = QSignalSpy(w.selectionFinished)
QTest.mousePress(view.viewport(),
Qt.LeftButton,
pos=pos + QPoint(1, 1))
mouseMove(view.viewport(), Qt.LeftButton, pos=pos + QPoint(20, 20))
QTest.mouseRelease(view.viewport(),
Qt.LeftButton,
pos=pos + QPoint(30, 40))
self.assertSequenceEqual(list(spy), [[]])
spy_fin = QSignalSpy(w.selectionFinished)
spy_chn = QSignalSpy(w.selectionChanged)
w.selectRows([1])
self.assertSequenceEqual(list(spy_fin), [])
self.assertSequenceEqual(list(spy_chn), [[]])
def test_colormap(self):
w = HeatmapGridWidget()
self.scene.addItem(w)
w.setHeatmaps(self._Data["2-2"])
w.setColorMap(GradientColorMap([[255] * 3, [0] * 3]))
w.setColorMap(GradientColorMap([[255] * 3, [0] * 3], center=0))