def render_drop_shadow_frame(pixmap, shadow_rect, shadow_color,
offset, radius, rect_fill_color):
pixmap.fill(QColor(0, 0, 0, 0))
scene = QGraphicsScene()
rect = QGraphicsRectItem(shadow_rect)
rect.setBrush(QColor(rect_fill_color))
rect.setPen(QPen(Qt.NoPen))
scene.addItem(rect)
effect = QGraphicsDropShadowEffect(color=shadow_color,
blurRadius=radius,
offset=offset)
rect.setGraphicsEffect(effect)
scene.setSceneRect(QRectF(QPointF(0, 0), QSizeF(pixmap.size())))
painter = QPainter(pixmap)
scene.render(painter)
painter.end()
scene.clear()
scene.deleteLater()
return pixmap
def render_drop_shadow_frame(pixmap, shadow_rect, shadow_color,
offset, radius, rect_fill_color):
pixmap.fill(QColor(0, 0, 0, 0))
scene = QGraphicsScene()
rect = QGraphicsRectItem(shadow_rect)
rect.setBrush(QColor(rect_fill_color))
rect.setPen(QPen(Qt.NoPen))
scene.addItem(rect)
effect = QGraphicsDropShadowEffect(color=shadow_color,
blurRadius=radius,
offset=offset)
rect.setGraphicsEffect(effect)
scene.setSceneRect(QRectF(QPointF(0, 0), QSizeF(pixmap.size())))
painter = QPainter(pixmap)
scene.render(painter)
painter.end()
scene.clear()
scene.deleteLater()
return pixmap
class TestItems(QAppTestCase):
def setUp(self):
super().setUp()
self.scene = QGraphicsScene()
self.view = QGraphicsView(self.scene)
self.view.setRenderHints(
QPainter.Antialiasing |
QPainter.SmoothPixmapTransform |
QPainter.TextAntialiasing
)
self.view.resize(500, 300)
self.view.show()
def tearDown(self):
self.scene.clear()
self.scene.deleteLater()
self.view.deleteLater()
del self.scene
del self.view
super().tearDown()
class TestItems(QAppTestCase):
def setUp(self):
super(TestItems, self).setUp()
self.scene = QGraphicsScene()
self.view = QGraphicsView(self.scene)
self.view.setRenderHints(
QPainter.Antialiasing |
QPainter.SmoothPixmapTransform |
QPainter.TextAntialiasing
)
self.view.resize(500, 300)
self.view.show()
def tearDown(self):
self.scene.clear()
self.scene.deleteLater()
self.view.deleteLater()
del self.scene
del self.view
super(TestItems, self).tearDown()
class TestItems(unittest.TestCase):
def setUp(self):
import logging
from AnyQt.QtWidgets import \
QApplication, QGraphicsScene, QGraphicsView
from AnyQt.QtGui import QPainter
from AnyQt.QtCore import QTimer
logging.basicConfig()
self.app = QApplication([])
self.scene = QGraphicsScene()
self.view = QGraphicsView(self.scene)
self.view.setRenderHints(
QPainter.Antialiasing | \
QPainter.SmoothPixmapTransform | \
QPainter.TextAntialiasing
)
self.view.resize(500, 300)
self.view.show()
QTimer.singleShot(10000, self.app.exit)
def my_excepthook(*args):
sys.setrecursionlimit(1010)
traceback.print_exc(limit=4)
self._orig_excepthook = sys.excepthook
sys.excepthook = my_excepthook
self.singleShot = QTimer.singleShot
def tearDown(self):
self.scene.clear()
self.scene.deleteLater()
self.view.deleteLater()
del self.scene
del self.view
self.app.processEvents()
del self.app
sys.excepthook = self._orig_excepthook
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))
