Spaces:
Running
Running
File size: 3,438 Bytes
8429e5e 61f1033 8429e5e 61f1033 8429e5e 61f1033 8429e5e 61f1033 8429e5e 61f1033 8429e5e 61f1033 8429e5e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | """Tests for gazet.geometry — coordinate rounding and geometry normalization."""
import json
from gazet.geometry import _round_coords, normalize_geometry_to_geojson
class TestRoundCoords:
def test_single_float(self):
assert _round_coords(1.23456789, 1) == 1.2
def test_zero_precision(self):
assert _round_coords(1.9, 0) == 2.0
def test_nested_list(self):
coords = [1.11111, 2.22222, 3.33333]
result = _round_coords(coords, 2)
assert result == [1.11, 2.22, 3.33]
def test_dict_recursion(self):
obj = {"lat": 1.99999, "lng": 2.11111}
result = _round_coords(obj, 1)
assert result == {"lat": 2.0, "lng": 2.1}
def test_nested_structure(self):
geojson = {
"type": "Point",
"coordinates": [12.123456, 34.987654],
}
result = _round_coords(geojson, 3)
assert result == {"type": "Point", "coordinates": [12.123, 34.988]}
def test_multipolygon(self):
coords = [[[1.1, 2.2], [3.3, 4.4], [5.5, 6.6]]]
result = _round_coords(coords, 1)
assert result == [[[1.1, 2.2], [3.3, 4.4], [5.5, 6.6]]]
def test_string_passthrough(self):
assert _round_coords("hello", 2) == "hello"
def test_int_passthrough(self):
assert _round_coords(42, 2) == 42
def test_none_passthrough(self):
assert _round_coords(None, 2) is None
def test_mixed_types_in_list(self):
data = [1.111, "type", 2.222]
result = _round_coords(data, 1)
assert result == [1.1, "type", 2.2]
class TestNormalizeGeometryToGeojson:
def test_no_geometry_column(self, con):
import pandas as pd
df = pd.DataFrame({"id": [1], "name": ["test"]})
result = normalize_geometry_to_geojson(con, df)
assert "geometry" not in result.columns
def test_empty_dataframe(self, con):
import pandas as pd
df = pd.DataFrame(columns=["geometry"])
result = normalize_geometry_to_geojson(con, df)
assert result.empty
def test_geojson_string_simplified(self, con):
import pandas as pd
# A simple point — simplify should pass it through
point_geojson = json.dumps(
{
"type": "Point",
"coordinates": [12.1234567, 34.9876543],
}
)
df = pd.DataFrame({"geometry": [point_geojson]})
result = normalize_geometry_to_geojson(con, df)
assert result["geometry"].iloc[0] is not None
parsed = json.loads(result["geometry"].iloc[0])
assert parsed["type"] == "Point"
# Coordinates should be rounded
coords = parsed["coordinates"]
assert coords[0] == round(12.1234567, 5)
assert coords[1] == round(34.9876543, 5)
def test_null_geometry_preserved(self, con):
import pandas as pd
df = pd.DataFrame({"geometry": [None, None]})
result = normalize_geometry_to_geojson(con, df)
assert pd.isna(result["geometry"]).all() or result["geometry"].isna().all()
def test_mixed_geojson_and_null(self, con):
import pandas as pd
point = json.dumps({"type": "Point", "coordinates": [0.0, 0.0]})
df = pd.DataFrame({"geometry": [point, None]})
result = normalize_geometry_to_geojson(con, df)
assert result["geometry"].iloc[0] is not None
assert result["geometry"].iloc[1] is None
|