1 files changed, 115 insertions, 0 deletions
diff --git a/sources/pyside6/tests/QtGui/qpainter_test.py b/sources/pyside6/tests/QtGui/qpainter_test.py
new file mode 100644
index 000000000..103b91ad6
--- /dev/null
+++ b/sources/pyside6/tests/QtGui/qpainter_test.py
@@ -0,0 +1,115 @@
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+import gc
+import os
+import sys
+import unittest
+
+from pathlib import Path
+sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
+from init_paths import init_test_paths
+init_test_paths(False)
+
+from helper.usesqapplication import UsesQApplication
+from PySide6.QtGui import QPainter, QLinearGradient, QImage
+from PySide6.QtCore import QLine, QLineF, QPoint, QPointF, QRect, QRectF, Qt
+
+
+try:
+    import numpy as np
+    HAVE_NUMPY = True
+except ModuleNotFoundError:
+    HAVE_NUMPY = False
+
+
+class QPainterDrawText(UsesQApplication):
+    def setUp(self):
+        super(QPainterDrawText, self).setUp()
+        self.image = QImage(32, 32, QImage.Format_ARGB32)
+        self.painter = QPainter(self.image)
+        self.text = 'teste!'
+
+    def tearDown(self):
+        del self.text
+        self.painter.end()
+        del self.painter
+        # PYSIDE-535: Need to collect garbage in PyPy to trigger deletion
+        gc.collect()
+        super(QPainterDrawText, self).tearDown()
+
+    def testDrawText(self):
+        # bug #254
+        rect = self.painter.drawText(100, 100, 100, 100,
+                                     Qt.AlignCenter | Qt.TextWordWrap,
+                                     self.text)
+        self.assertTrue(isinstance(rect, QRect))
+
+    def testDrawTextWithRect(self):
+        # bug #225
+        rect = QRect(100, 100, 100, 100)
+        newRect = self.painter.drawText(rect, Qt.AlignCenter | Qt.TextWordWrap,
+                                        self.text)
+
+        self.assertTrue(isinstance(newRect, QRect))
+
+    def testDrawTextWithRectF(self):
+        '''QPainter.drawText(QRectF, ... ,QRectF*) inject code'''
+        rect = QRectF(100, 52.3, 100, 100)
+        newRect = self.painter.drawText(rect, Qt.AlignCenter | Qt.TextWordWrap,
+                                        self.text)
+
+        self.assertTrue(isinstance(newRect, QRectF))
+
+    def testDrawOverloads(self):
+        '''Calls QPainter.drawLines overloads, if something is
+           wrong Exception and chaos ensues. Bug #395'''
+        self.painter.drawLines([QLine(QPoint(0, 0), QPoint(1, 1))])
+        self.painter.drawLines([QPoint(0, 0), QPoint(1, 1)])
+        self.painter.drawLines([QPointF(0, 0), QPointF(1, 1)])
+        self.painter.drawLines([QLineF(QPointF(0, 0), QPointF(1, 1))])
+        self.painter.drawPoints([QPoint(0, 0), QPoint(1, 1)])
+        self.painter.drawPoints([QPointF(0, 0), QPointF(1, 1)])
+        self.painter.drawConvexPolygon([QPointF(10.0, 80.0),
+                                        QPointF(20.0, 10.0),
+                                        QPointF(80.0, 30.0),
+                                        QPointF(90.0, 70.0)])
+        self.painter.drawConvexPolygon([QPoint(10.0, 80.0),
+                                        QPoint(20.0, 10.0),
+                                        QPoint(80.0, 30.0),
+                                        QPoint(90.0, 70.0)])
+        self.painter.drawPolygon([QPointF(10.0, 80.0),
+                                  QPointF(20.0, 10.0),
+                                  QPointF(80.0, 30.0),
+                                  QPointF(90.0, 70.0)])
+        self.painter.drawPolygon([QPoint(10.0, 80.0),
+                                  QPoint(20.0, 10.0),
+                                  QPoint(80.0, 30.0),
+                                  QPoint(90.0, 70.0)])
+        self.painter.drawPolyline([QPointF(10.0, 80.0),
+                                   QPointF(20.0, 10.0),
+                                   QPointF(80.0, 30.0),
+                                   QPointF(90.0, 70.0)])
+        self.painter.drawPolyline([QPoint(10.0, 80.0),
+                                   QPoint(20.0, 10.0),
+                                   QPoint(80.0, 30.0),
+                                   QPoint(90.0, 70.0)])
+        if HAVE_NUMPY:
+            x = np.array([10.0, 20.0, 80.0, 90.0])
+            y = np.array([80.0, 10.0, 30.0, 70.0])
+            self.painter.drawPointsNp(x, y)
+
+
+class SetBrushWithOtherArgs(UsesQApplication):
+    '''Using qpainter.setBrush with args other than QBrush'''
+
+    def testSetBrushGradient(self):
+        image = QImage(32, 32, QImage.Format_ARGB32)
+        with QPainter(image) as painter:
+            gradient = QLinearGradient(0, 0, 0, 0)
+            painter.setBrush(gradient)
+
+
+if __name__ == '__main__':
+    unittest.main()
+