test_wkt.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. from math import pi
  2. import pytest
  3. from shapely.geometry import Point
  4. from shapely.wkt import dump, dumps, load, loads
  5. @pytest.fixture(scope="module")
  6. def some_point():
  7. return Point(pi, -pi)
  8. @pytest.fixture(scope="module")
  9. def empty_geometry():
  10. return Point()
  11. def test_wkt(some_point):
  12. """.wkt and wkt.dumps() both do not trim by default."""
  13. assert some_point.wkt == f"POINT ({pi:.15f} {-pi:.15f})"
  14. def test_wkt_null(empty_geometry):
  15. assert empty_geometry.wkt == "POINT EMPTY"
  16. def test_dump_load(some_point, tmpdir):
  17. file = tmpdir.join("test.wkt")
  18. with open(file, "w") as file_pointer:
  19. dump(some_point, file_pointer)
  20. with open(file) as file_pointer:
  21. restored = load(file_pointer)
  22. assert some_point == restored
  23. def test_dump_load_null_geometry(empty_geometry, tmpdir):
  24. file = tmpdir.join("test.wkt")
  25. with open(file, "w") as file_pointer:
  26. dump(empty_geometry, file_pointer)
  27. with open(file) as file_pointer:
  28. restored = load(file_pointer)
  29. # This is does not work with __eq__():
  30. assert empty_geometry.equals(restored)
  31. def test_dumps_loads(some_point):
  32. assert dumps(some_point) == f"POINT ({pi:.16f} {-pi:.16f})"
  33. assert loads(dumps(some_point)) == some_point
  34. def test_dumps_loads_null_geometry(empty_geometry):
  35. assert dumps(empty_geometry) == "POINT EMPTY"
  36. # This is does not work with __eq__():
  37. assert loads(dumps(empty_geometry)).equals(empty_geometry)
  38. def test_dumps_precision(some_point):
  39. assert dumps(some_point, rounding_precision=4) == f"POINT ({pi:.4f} {-pi:.4f})"