| import math |
| import unittest |
|
|
| from src.geometry import ( |
| calculate_site_metrics, |
| normalize_map_state, |
| polygon_area_local, |
| polygon_perimeter_local, |
| ) |
|
|
|
|
| class GeometryTests(unittest.TestCase): |
| def test_local_polygon_area_and_perimeter(self): |
| points = [(0, 0), (10, 0), (10, 5), (0, 5), (0, 0)] |
|
|
| self.assertEqual(polygon_area_local(points), 50) |
| self.assertEqual(polygon_perimeter_local(points), 30) |
|
|
| def test_drawn_polygon_normalizes_to_site_selection(self): |
| map_state = { |
| "mode": "polygon", |
| "geometry": { |
| "type": "Polygon", |
| "coordinates": [ |
| [ |
| [70.0, 21.0], |
| [70.002, 21.0], |
| [70.002, 21.002], |
| [70.0, 21.002], |
| [70.0, 21.0], |
| ] |
| ], |
| }, |
| } |
|
|
| selection = normalize_map_state(map_state) |
|
|
| self.assertEqual(selection.selection_type, "drawn_polygon") |
| self.assertEqual(selection.coordinate_mode, "wgs84") |
| self.assertIn("approximate", selection.accuracy_label) |
| self.assertGreater(selection.area_sqm, 0) |
| self.assertGreater(selection.perimeter_m, 0) |
|
|
| def test_pin_radius_normalizes_to_site_selection(self): |
| selection = normalize_map_state( |
| {"mode": "pin_radius", "lat": 21.0, "lon": 70.0, "radius_m": 250} |
| ) |
|
|
| self.assertEqual(selection.selection_type, "pin_radius") |
| self.assertEqual(selection.radius_m, 250) |
| self.assertAlmostEqual(selection.area_sqm, math.pi * 250 * 250, delta=1) |
| self.assertIn("not exact", " ".join(selection.limitations)) |
|
|
| def test_calculate_site_metrics_rejects_empty_geometry(self): |
| with self.assertRaises(ValueError): |
| calculate_site_metrics({"type": "Polygon", "coordinates": []}) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
