62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
from math import pi
|
|
|
|
import pytest
|
|
|
|
from shapely.geometry import Point
|
|
from shapely.wkt import dump, dumps, load, loads
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def some_point():
|
|
return Point(pi, -pi)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def empty_geometry():
|
|
return Point()
|
|
|
|
|
|
def test_wkt(some_point):
|
|
""".wkt and wkt.dumps() both do not trim by default."""
|
|
assert some_point.wkt == f"POINT ({pi:.15f} {-pi:.15f})"
|
|
|
|
|
|
def test_wkt_null(empty_geometry):
|
|
assert empty_geometry.wkt == "POINT EMPTY"
|
|
|
|
|
|
def test_dump_load(some_point, tmpdir):
|
|
file = tmpdir.join("test.wkt")
|
|
with open(file, "w") as file_pointer:
|
|
dump(some_point, file_pointer)
|
|
with open(file) as file_pointer:
|
|
restored = load(file_pointer)
|
|
|
|
assert some_point == restored
|
|
|
|
|
|
def test_dump_load_null_geometry(empty_geometry, tmpdir):
|
|
file = tmpdir.join("test.wkt")
|
|
with open(file, "w") as file_pointer:
|
|
dump(empty_geometry, file_pointer)
|
|
with open(file) as file_pointer:
|
|
restored = load(file_pointer)
|
|
|
|
# This is does not work with __eq__():
|
|
assert empty_geometry.equals(restored)
|
|
|
|
|
|
def test_dumps_loads(some_point):
|
|
assert dumps(some_point) == f"POINT ({pi:.16f} {-pi:.16f})"
|
|
assert loads(dumps(some_point)) == some_point
|
|
|
|
|
|
def test_dumps_loads_null_geometry(empty_geometry):
|
|
assert dumps(empty_geometry) == "POINT EMPTY"
|
|
# This is does not work with __eq__():
|
|
assert loads(dumps(empty_geometry)).equals(empty_geometry)
|
|
|
|
|
|
def test_dumps_precision(some_point):
|
|
assert dumps(some_point, rounding_precision=4) == f"POINT ({pi:.4f} {-pi:.4f})"
|