iwyu

Author: yohanboniface

tests/asyncpg/test_geometrycollection.py

@@ -0,0 +1,27 @@
+import pytest
+
+from postgis import Point, LineString, Polygon, GeometryCollection
+
+pytestmark = pytest.mark.asyncio
+
+
+POLYGON = Polygon((
+    ((1, 2), (3, 4), (5, 6), (1, 2)),
+    ((2, 3), (4, 5), (6, 7), (2, 3))
+))
+
+COLLECTION = [
+    Point(1, 2),
+    LineString(((1, 2), (3, 4))),
+    POLYGON
+]
+
+
+async def test_geometrycollection_should_round(connection):
+    geom = GeometryCollection(COLLECTION, srid=4326)
+    await connection.execute('INSERT INTO geometrycollection_async (geom) '
+                             'VALUES ($1)', geom)
+    geom = await connection.fetchval('SELECT geom '
+                                     'FROM geometrycollection_async '
+                                     'WHERE geom=$1', geom, column=0)
+    assert geom == COLLECTION

tests/asyncpg/test_linestring.py

@@ -0,0 +1,17 @@
+import pytest
+
+from postgis import LineString
+
+pytestmark = pytest.mark.asyncio
+
+
+@pytest.mark.parametrize('expected', [
+    ((30, 10), (10, 30), (40, 40)),
+])
+async def test_linestring_should_round(connection, expected):
+    geom = LineString(expected, srid=4326)
+    await connection.execute('INSERT INTO linestring_async (geom) VALUES ($1)',
+                             geom)
+    geom = await connection.fetchval('SELECT geom FROM linestring_async WHERE '
+                                     'geom=$1', geom, column=0)
+    assert geom.coords == expected

tests/asyncpg/test_multilinestring.py

@@ -0,0 +1,17 @@
+import pytest
+
+from postgis import MultiLineString
+
+pytestmark = pytest.mark.asyncio
+
+
+@pytest.mark.parametrize('expected', [
+    (((30, 10), (10, 30)), ((40, 10), (10, 40))),
+])
+async def test_multilinestring_should_round(connection, expected):
+    geom = MultiLineString(expected, srid=4326)
+    await connection.execute('INSERT INTO multilinestring_async (geom) '
+                             'VALUES ($1)', geom)
+    geom = await connection.fetchval('SELECT geom FROM multilinestring_async '
+                                     'WHERE geom=$1', geom, column=0)
+    assert geom.coords == expected

tests/asyncpg/test_multipolygon.py

@@ -0,0 +1,28 @@
+import pytest
+
+from postgis import MultiPolygon
+
+pytestmark = pytest.mark.asyncio
+
+MULTI = (
+            (
+                ((35, 10), (45, 45), (15, 40), (10, 20), (35, 10)),
+                ((20, 30), (35, 35), (30, 20), (20, 30))
+            ),
+            (
+                ((36, 10), (46, 45), (16, 40), (16, 20), (36, 10)),
+                ((21, 30), (36, 35), (36, 20), (21, 30))
+            ),
+        )
+
+
+@pytest.mark.parametrize('expected', [
+    MULTI,
+])
+async def test_multipolygon_should_round(connection, expected):
+    geom = MultiPolygon(expected, srid=4326)
+    await connection.execute('INSERT INTO multipolygon_async (geom) '
+                             'VALUES ($1)', geom)
+    geom = await connection.fetchval('SELECT geom FROM multipolygon_async '
+                                     'WHERE geom=$1', geom, column=0)
+    assert geom.coords == expected

tests/asyncpg/test_polygon.py

@@ -0,0 +1,17 @@
+import pytest
+
+from postgis import Polygon
+
+pytestmark = pytest.mark.asyncio
+
+
+@pytest.mark.parametrize('expected', [
+    (((35, 10), (45, 45), (15, 40), (10, 20), (35, 10)), ((20, 30), (35, 35), (30, 20), (20, 30))),  # noqa
+])
+async def test_polygon_should_round(connection, expected):
+    geom = Polygon(expected, srid=4326)
+    await connection.execute('INSERT INTO polygon_async (geom) VALUES ($1)',
+                             geom)
+    geom = await connection.fetchval('SELECT geom FROM polygon_async WHERE '
+                                     'geom=$1', geom, column=0)
+    assert geom.coords == expected