mirror of
https://github.com/qgis/QGIS.git
synced 2025-10-05 00:09:32 -04:00
tests: Add python tests for QgsSfcgalGeometry
This commit is contained in:
Jean Felder
Nyall Dawson
parent
22be110457
commit
4340ce8ae6
@ -693,3 +693,7 @@ if (WITH_SERVER)
|
||||
ADD_PYTHON_TEST(PyQgsServerConfigCache test_qgsserver_configcache.py)
|
||||
ADD_PYTHON_TEST(PyQgsServerWMSGetCapabilitiesGroupName test_qgsserver_wms_getcapabilities_group_name.py)
|
||||
endif()
|
||||
|
||||
if (WITH_SFCGAL)
|
||||
ADD_PYTHON_TEST(PyQgsSfcgal test_qgssfcgal.py)
|
||||
endif()
|
||||
|
||||
285
tests/src/python/test_qgssfcgal.py
Normal file
285
tests/src/python/test_qgssfcgal.py
Normal file
@ -0,0 +1,285 @@
|
||||
"""QGIS Unit tests for QgsSfcgalGeometry
|
||||
|
||||
.. note:: This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; either version 2 of the License, or
|
||||
(at your option) any later version.
|
||||
"""
|
||||
|
||||
__author__ = "Jean Felder"
|
||||
__date__ = "05/08/2025"
|
||||
__copyright__ = "Copyright 2025, The QGIS Project"
|
||||
|
||||
import unittest
|
||||
from math import pi
|
||||
|
||||
from qgis.core import (
|
||||
Qgis,
|
||||
QgsConstWkbPtr,
|
||||
QgsPoint,
|
||||
QgsPolyhedralSurface,
|
||||
QgsSfcgalGeometry,
|
||||
QgsVector3D,
|
||||
)
|
||||
from qgis.testing import start_app, QgisTestCase
|
||||
|
||||
start_app()
|
||||
|
||||
|
||||
class TestQgsSFCGAL(QgisTestCase):
|
||||
|
||||
def test_constructor(self):
|
||||
default_geom = QgsSfcgalGeometry(QgsPolyhedralSurface())
|
||||
is_empty, err_msg = default_geom.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(is_empty)
|
||||
|
||||
surface = QgsPolyhedralSurface()
|
||||
surface_wkt = "POLYHEDRALSURFACE Z (((0 0 0,0 1 0,1 1 0,0 0 0)))"
|
||||
surface.fromWkt(surface_wkt)
|
||||
self.assertFalse(surface.isEmpty())
|
||||
self.assertEqual(surface.numPatches(), 1)
|
||||
self.assertTrue(surface.is3D())
|
||||
geom_surface = QgsSfcgalGeometry(surface)
|
||||
is_empty, err_msg = geom_surface.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(is_empty)
|
||||
wkt_out, err_msg = geom_surface.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, surface_wkt)
|
||||
|
||||
point = QgsSfcgalGeometry(QgsPoint(2, 3))
|
||||
is_empty, err_msg = point.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(is_empty)
|
||||
is_valid, err_msg = point.isValid()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(is_valid)
|
||||
wkt_out, err_msg = point.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, "POINT (2 3)")
|
||||
|
||||
def test_from_wkt(self):
|
||||
tin_wkt = "TIN M (((0 0 3,0 1 3,1 1 3,0 0 3)))"
|
||||
geom, err_msg = QgsSfcgalGeometry.fromWkt(tin_wkt)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(isinstance(geom, QgsSfcgalGeometry))
|
||||
is_empty, err_msg = geom.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(is_empty)
|
||||
|
||||
wkt_out, err_msg = geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, tin_wkt)
|
||||
|
||||
def test_from_wkb(self):
|
||||
wkb_in = b"\x01\xf8\x03\x00\x00\x02\x00\x00\x00\x01\xf9\x03\x00\x00\x01\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\xf9\x03\x00\x00\x01\x00\x00\x00\x04\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
||||
geom, err_msg = QgsSfcgalGeometry.fromWkb(QgsConstWkbPtr(wkb_in))
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(isinstance(geom, QgsSfcgalGeometry))
|
||||
is_empty, err_msg = geom.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(is_empty)
|
||||
|
||||
expected_wkt = "TIN Z (((0 0 0,1 0 0,0 1 0,0 0 0)),((1 0 0,1 1 0,0 1 0,1 0 0)))"
|
||||
wkt_out, err_msg = geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
wkb_out, err_msg = geom.asWkb()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkb_out, wkb_in)
|
||||
|
||||
def test_is_valid(self):
|
||||
surface_wkt = "POLYHEDRALSURFACE Z (((0 0 0,0 100 0,100 100 0,100 0 0,0 0 0)),((0 0 0,50 50 50,0 100 0,0 0 0)),((100 0 0,100 100 0,50 50 50,100 0 0)))"
|
||||
sfcgal_geom, err_msg = QgsSfcgalGeometry.fromWkt(surface_wkt)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
is_empty, err_msg = sfcgal_geom.isEmpty()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(is_empty)
|
||||
|
||||
is_valid, err_msg = sfcgal_geom.isValid()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(is_valid)
|
||||
|
||||
def test_part_count(self):
|
||||
geoms = [
|
||||
["TIN Z (((0 0 0,1 0 0,0 1 0,0 0 0)),((1 0 0,1 1 0,0 1 0,1 0 0)))", 2],
|
||||
["POINT (0 1)", 1],
|
||||
[
|
||||
"GEOMETRYCOLLECTION (POINT (0 0),POINT (2 2),LINESTRING (1 1,2 2,1 1))",
|
||||
3,
|
||||
],
|
||||
["TRIANGLE ((1 1,2 2,2 1,1 1))", 3],
|
||||
[
|
||||
"MULTIPOINT ((50 40),(140 70),(80 100),(130 140),(30 150),(70 180),(190 110),(120 20))",
|
||||
8,
|
||||
],
|
||||
[
|
||||
"POLYGON ((100 180,160 20,20 20,100 180),(100 180,80 60,120 60,100 180))",
|
||||
2,
|
||||
],
|
||||
[
|
||||
"POLYHEDRALSURFACE Z (((0 0 0,0 100 0,100 100 0,100 0 0,0 0 0)),((0 0 0,50 50 50,0 100 0,0 0 0)),((100 0 0,100 100 0,50 50 50,100 0 0)))",
|
||||
3,
|
||||
],
|
||||
]
|
||||
for geom in geoms:
|
||||
sfcgal_geom, err_msg = QgsSfcgalGeometry.fromWkt(geom[0])
|
||||
self.assertFalse(err_msg)
|
||||
wkt_out, err_msg = sfcgal_geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, geom[0])
|
||||
|
||||
part_count, err_msg = sfcgal_geom.partCount()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(part_count, geom[1])
|
||||
|
||||
def test_area(self):
|
||||
geoms = [
|
||||
["TRIANGLE ((1 1,2 2,2 1,1 1))", 0.5],
|
||||
[
|
||||
"POLYGON ((100 180,160 20,20 20,100 180),(100 180,80 60,120 60,100 180))",
|
||||
8800.0,
|
||||
],
|
||||
["LINESTRING (0 0,1 1,2 2,0 0)", 0.0],
|
||||
["LINESTRING Z (0 0 0,1 0 0,0 1 0,0 0 1,0 0 0)", 0.0],
|
||||
[
|
||||
"POLYHEDRALSURFACE Z (((0 0 0,0 100 0,100 100 0,100 0 0,0 0 0)),((0 0 0,50 50 50,0 100 0,0 0 0)),((100 0 0,100 100 0,50 50 50,100 0 0)))",
|
||||
17071.06781,
|
||||
],
|
||||
]
|
||||
for geom in geoms:
|
||||
sfcgal_geom, err_msg = QgsSfcgalGeometry.fromWkt(geom[0])
|
||||
|
||||
self.assertFalse(err_msg)
|
||||
wkt_out, err_msg = sfcgal_geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, geom[0])
|
||||
|
||||
area, err_msg = sfcgal_geom.area()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertAlmostEqual(area, geom[1], 5)
|
||||
|
||||
def test_scale(self):
|
||||
wkt = "LINESTRING Z (0 0 0,1 0 0,0 1 0,0 0 1,0 0 0)"
|
||||
sfcgal_geom, err_msg = QgsSfcgalGeometry.fromWkt(wkt)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
scaled_geom, err_msg = sfcgal_geom.scale(QgsVector3D(1.5, 2, 0.5))
|
||||
self.assertFalse(err_msg)
|
||||
expected_wkt = (
|
||||
"LINESTRING Z (0.0 0.0 0.0,1.5 0.0 0.0,0.0 2.0 0.0,0.0 0.0 0.5,0.0 0.0 0.0)"
|
||||
)
|
||||
wkt_out, err_msg = scaled_geom.asWkt(1)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
def test_rotate(self):
|
||||
# 2D rotation
|
||||
wkt = "TRIANGLE ((1 1,2 2,2 1,1 1))"
|
||||
sfcgal_geom, err_msg = QgsSfcgalGeometry.fromWkt(wkt)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
rotated_2d_geom, err_msg = sfcgal_geom.rotate2D(pi / 2.0, QgsPoint(1, 1))
|
||||
self.assertFalse(err_msg)
|
||||
expected_wkt = "TRIANGLE ((1 1,0 2,1 2,1 1))"
|
||||
wkt_out, err_msg = rotated_2d_geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
# 3D rotation
|
||||
rotated_3d_geom, err_msg = sfcgal_geom.rotate3D(
|
||||
pi / 2.0, QgsVector3D(2, 1, 0), QgsPoint(1, 1, 0)
|
||||
)
|
||||
self.assertFalse(err_msg)
|
||||
expected_wkt = "TRIANGLE Z ((1 1 0,2 2 0,2 1 0,1 1 0))"
|
||||
wkt_out, err_msg = rotated_3d_geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
def test_intersection(self):
|
||||
point = QgsSfcgalGeometry(QgsPoint(2, 3))
|
||||
|
||||
polygon1, err_msg = QgsSfcgalGeometry.fromWkt(
|
||||
"POLYGON ((0 0,10 0,10 10,0 10,0 0))"
|
||||
)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
intersects, err_msg = polygon1.intersects(point)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(intersects)
|
||||
|
||||
polygon2, err_msg = QgsSfcgalGeometry.fromWkt(
|
||||
"POLYGON ((-1 -1,1 -1,1 1,-1 1,-1 -1))"
|
||||
)
|
||||
self.assertFalse(err_msg)
|
||||
intersects, err_msg = polygon2.intersects(point)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertFalse(intersects)
|
||||
|
||||
intersects, err_msg = polygon2.intersects(polygon1)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertTrue(intersects)
|
||||
|
||||
intersection, err_msg = polygon2.intersection(polygon1)
|
||||
self.assertFalse(err_msg)
|
||||
wkt_out, err_msg = intersection.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
expected_intersection = "POLYGON ((0 1,0 0,1 0,1 1,0 1))"
|
||||
self.assertEqual(wkt_out, expected_intersection)
|
||||
|
||||
def test_combine(self):
|
||||
point = QgsSfcgalGeometry(QgsPoint(2, 3))
|
||||
|
||||
combined_geom, err_msg = point.combine([QgsPoint(4, 5)])
|
||||
self.assertFalse(err_msg)
|
||||
wkt_out, err_msg = combined_geom.asWkt(0)
|
||||
expected_wkt = "MULTIPOINT ((2 3),(4 5))"
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
def test_difference(self):
|
||||
polygon1, err_msg = QgsSfcgalGeometry.fromWkt(
|
||||
"POLYGON ((0 0,10 0,10 10,0 10,0 0))"
|
||||
)
|
||||
self.assertFalse(err_msg)
|
||||
polygon2, err_msg = QgsSfcgalGeometry.fromWkt(
|
||||
"POLYGON ((-1 -1,1 -1,1 1,-1 1,-1 -1))"
|
||||
)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
diff_geom, err_msg = polygon1.difference(polygon2)
|
||||
self.assertFalse(err_msg)
|
||||
wkt_out, err_msg = diff_geom.asWkt(0)
|
||||
expected_wkt = "POLYGON ((0 10,0 1,1 1,1 0,10 0,10 10,0 10))"
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
def test_triangulate(self):
|
||||
poly_wkt = "POLYGON ((0.0 0.0,1.0 0.0,1.0 1.0,0.0 1.0,0.0 0.0),(0.2 0.2,0.2 0.8,0.8 0.8,0.8 0.2,0.2 0.2))"
|
||||
polygon, err_msg = QgsSfcgalGeometry.fromWkt(poly_wkt)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
triangulation, err_msg = polygon.triangulate()
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(triangulation.wkbType()[0], Qgis.WkbType.TIN)
|
||||
self.assertEqual(triangulation.partCount()[0], 10)
|
||||
|
||||
def test_convexhull(self):
|
||||
poly_wkt = "POLYGON ((0.0 0.0,1.0 0.0,1.0 1.0,0.0 1.0,0.0 0.0),(0.2 0.2,0.2 0.8,0.8 0.8,0.8 0.2,0.2 0.2))"
|
||||
polygon, err_msg = QgsSfcgalGeometry.fromWkt(poly_wkt)
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
convex_hull, err_msg = polygon.convexHull()
|
||||
self.assertFalse(err_msg)
|
||||
|
||||
expected_wkt = "POLYGON ((0 0,1 0,1 1,0 1,0 0))"
|
||||
wkt_out, err_msg = diff_geom.asWkt(0)
|
||||
self.assertFalse(err_msg)
|
||||
self.assertEqual(wkt_out, expected_wkt)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Loading…
x
Reference in New Issue
Block a user