| import unittest |
|
|
| from src.earth_reference import build_earth_reference |
| from src.models import SiteSelection |
| from src.safety import find_forbidden_phrases |
|
|
|
|
| class EarthReferenceTests(unittest.TestCase): |
| def test_earth_reference_generates_external_links_and_safe_evidence(self): |
| selection = SiteSelection( |
| id="S1", |
| selection_type="drawn_polygon", |
| coordinate_mode="wgs84", |
| geometry_geojson=None, |
| local_geometry=None, |
| anchor_lat=21.00245, |
| anchor_lon=70.24550, |
| radius_m=None, |
| area_sqm=1000, |
| perimeter_m=150, |
| centroid=(21.00245, 70.24550), |
| bbox=None, |
| unit_source="WGS84", |
| accuracy_label="user-drawn approximate boundary", |
| limitations=["Drawn boundaries are approximate."], |
| ) |
|
|
| reference, evidence = build_earth_reference(selection, "Visible trees along the west edge.") |
|
|
| self.assertIn("Google Maps reference", reference["links"]) |
| self.assertIn("Google Earth reference", reference["links"]) |
| self.assertIn("visual-reference workflow", reference["markdown"]) |
| self.assertIn("Visible trees", reference["markdown"]) |
| self.assertGreaterEqual(len(evidence), 2) |
| self.assertEqual(find_forbidden_phrases(reference["markdown"]), []) |
|
|
| def test_earth_reference_without_anchor_is_explicitly_limited(self): |
| selection = SiteSelection( |
| id="S2", |
| selection_type="dxf_boundary", |
| coordinate_mode="local_cad", |
| geometry_geojson=None, |
| local_geometry=[(0, 0), (10, 0), (10, 10)], |
| anchor_lat=None, |
| anchor_lon=None, |
| radius_m=None, |
| area_sqm=100, |
| perimeter_m=40, |
| centroid=(5, 5), |
| bbox=(0, 0, 10, 10), |
| unit_source="DXF meters", |
| accuracy_label="cad local without anchor", |
| limitations=["CAD needs anchor."], |
| ) |
|
|
| reference, evidence = build_earth_reference(selection, "") |
|
|
| self.assertEqual(reference["links"], {}) |
| self.assertIn("no anchor coordinate", reference["markdown"]) |
| self.assertEqual(evidence, []) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
