diff --git a/tests/src/python/test_qgsgeometry.py b/tests/src/python/test_qgsgeometry.py index f7be7c1435e..fc606f631fc 100644 --- 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 @@ class TestQgsGeometry(unittest.TestCase): 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__':