Added from wkbPoint and Polyline tests to python geometry tests

qgis · Aug 3, 2012 · 8761bc7 · 8761bc7
1 parent 5ae312c
commit 8761bc7
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/tests/src/python/test_qgsgeometry.py b/tests/src/python/test_qgsgeometry.py
@@ -1,6 +1,8 @@
 import unittest
 
-from qgis.core import (QgsGeometry, QGis)
+from qgis.core import (QgsGeometry,
+                       QgsPoint,
+                       QGis)
 
 # Convenience instances in case you may need them
 # not used in this test
@@ -13,8 +15,20 @@ def testWktPointLoading(self):
         myWKT='POINT(10 10)'
         myGeometry = QgsGeometry.fromWkt(myWKT)
         myMessage = ('Expected:\n%s\nGot:\n%s\n' %
-                      (QGis.Point, myGeometry.type()))
-        assert myGeometry.type() == QGis.Point, myMessage
+                      (QGis.WKBPoint, myGeometry.type()))
+        assert myGeometry.wkbType() == QGis.WKBPoint, myMessage
+
+    def testFromPoint(self):
+        myPoint = QgsGeometry.fromPoint(QgsPoint(10, 10))
+        myMessage = ('Expected:\n%s\nGot:\n%s\n' %
+                      (QGis.WKBPoint, myPoint.type()))
+        assert myPoint.wkbType() == QGis.WKBPoint, myMessage
+
+    def testFromLine(self):
+        myLine = QgsGeometry.fromPolyline([QgsPoint(1, 1), QgsPoint(2, 2)])
+        myMessage = ('Expected:\n%s\nGot:\n%s\n' %
+                      (QGis.WKBLineString, myLine.type()))
+        assert myLine.wkbType() == QGis.WKBLineString, myMessage
 
 
 if __name__ == '__main__':