1 files changed, 152 insertions, 0 deletions
diff --git a/tests/QtGui/qgraphicsscene_test.py b/tests/QtGui/qgraphicsscene_test.py
new file mode 100644
index 000000000..98177e75a
--- /dev/null
+++ b/tests/QtGui/qgraphicsscene_test.py
@@ -0,0 +1,152 @@
+
+'''Basic test cases for QGraphicsScene'''
+
+import unittest
+import gc
+
+from PySide.QtCore import QPointF
+from PySide.QtGui import QApplication, QPushButton, QPolygonF, QPixmap
+from PySide.QtGui import QGraphicsScene, QPainterPath
+from PySide.QtGui import QGraphicsEllipseItem, QGraphicsLineItem
+from PySide.QtGui import QGraphicsPathItem, QGraphicsPixmapItem
+from PySide.QtGui import QGraphicsPolygonItem, QGraphicsRectItem
+from PySide.QtGui import QGraphicsSimpleTextItem, QGraphicsTextItem
+from PySide.QtGui import QGraphicsProxyWidget
+
+from helper import UsesQApplication
+
+class Constructor(unittest.TestCase):
+    '''QGraphicsScene constructor'''
+
+    def testConstructor(self):
+        #QGraphicsScene constructor
+        obj = QGraphicsScene()
+        self.assert_(isinstance(obj, QGraphicsScene))
+
+
+class ConstructorWithRect(unittest.TestCase):
+    '''QGraphicsScene qrect constructor and related sizes'''
+
+    def setUp(self):
+        #Acquire resources
+        # PyQt4 doesn't accept a QRect as argument to constructor
+        self.scene = QGraphicsScene(0, 200, 150, 175)
+
+    def tearDown(self):
+        #Release resources
+        del self.scene
+
+    def testHeight(self):
+        #QGraphicsScene.height()
+        self.assertEqual(self.scene.height(), 175)
+
+    def testWidth(self):
+        #QGraphicsScene.width()
+        self.assertEqual(self.scene.width(), 150)
+
+
+class AddItem(UsesQApplication):
+    '''Tests for QGraphicsScene.add*'''
+
+    qapplication = True
+
+    def setUp(self):
+        #Acquire resources
+        super(AddItem, self).setUp()
+        self.scene = QGraphicsScene()
+        # While the scene does not inherits from QWidget, requires
+        # an application to make the internals work.
+
+    def tearDown(self):
+        #Release resources
+        del self.scene
+        super(AddItem, self).tearDown()
+
+    def testEllipse(self):
+        #QGraphicsScene.addEllipse
+        item = self.scene.addEllipse(100, 100, 100, 100)
+        self.assert_(isinstance(item, QGraphicsEllipseItem))
+
+    def testLine(self):
+        #QGraphicsScene.addLine
+        item = self.scene.addLine(100, 100, 200, 200)
+        self.assert_(isinstance(item, QGraphicsLineItem))
+
+    def testPath(self):
+        #QGraphicsScene.addPath
+        item = self.scene.addPath(QPainterPath())
+        self.assert_(isinstance(item, QGraphicsPathItem))
+
+    def testPixmap(self):
+        #QGraphicsScene.addPixmap
+        item = self.scene.addPixmap(QPixmap())
+        self.assert_(isinstance(item, QGraphicsPixmapItem))
+
+    def testPolygon(self):
+        #QGraphicsScene.addPolygon
+        points = [QPointF(0, 0), QPointF(100, 100), QPointF(0, 100)]
+        item = self.scene.addPolygon(QPolygonF(points))
+        self.assert_(isinstance(item, QGraphicsPolygonItem))
+
+    def testRect(self):
+        #QGraphicsScene.addRect
+        item = self.scene.addRect(100, 100, 100, 100)
+        self.assert_(isinstance(item, QGraphicsRectItem))
+
+    def testSimpleText(self):
+        #QGraphicsScene.addSimpleText
+        item = self.scene.addSimpleText('Monty Python 42')
+        self.assert_(isinstance(item, QGraphicsSimpleTextItem))
+
+    def testText(self):
+        #QGraphicsScene.addText
+        item = self.scene.addText('Monty Python 42')
+        self.assert_(isinstance(item, QGraphicsTextItem))
+
+    def testWidget(self):
+        #QGraphicsScene.addWidget
+        # XXX: printing some X11 error when using under PyQt4
+        item = self.scene.addWidget(QPushButton())
+        self.assert_(isinstance(item, QGraphicsProxyWidget))
+
+
+class ItemRetrieve(UsesQApplication):
+    '''Tests for QGraphicsScene item retrieval methods'''
+
+    qapplication = True
+
+    def setUp(self):
+        #Acquire resources
+        super(ItemRetrieve, self).setUp()
+        self.scene = QGraphicsScene()
+
+        self.topleft = QGraphicsRectItem(0, 0, 100, 100)
+        self.topright = QGraphicsRectItem(100, 0, 100, 100)
+        self.bottomleft = QGraphicsRectItem(0, 100, 100, 100)
+        self.bottomright = QGraphicsRectItem(100, 100, 100, 100)
+
+        self.items = [self.topleft, self.topright, self.bottomleft,
+                        self.bottomright]
+
+        for item in self.items:
+            self.scene.addItem(item)
+
+    def tearDown(self):
+        #Release resources
+        del self.scene
+        super(ItemRetrieve, self).tearDown()
+
+    def testItems(self):
+        #QGraphicsScene.items()
+        items = self.scene.items()
+        self.assertEqual(items.sort(), self.items.sort())
+
+    def testItemAt(self):
+        #QGraphicsScene.itemAt()
+        self.assertEqual(self.scene.itemAt(50, 50), self.topleft)
+        self.assertEqual(self.scene.itemAt(150, 50), self.topright)
+        self.assertEqual(self.scene.itemAt(50, 150), self.bottomleft)
+        self.assertEqual(self.scene.itemAt(150, 150), self.bottomright)
+
+if __name__ == '__main__':
+    unittest.main()