| import unittest |
|
|
| from src.evidence import make_evidence |
| from src.models import SiteSelection |
| from src.safety import find_forbidden_phrases |
| from src.small_model_assistant import build_assistant_brief |
|
|
|
|
| class SmallModelAssistantTests(unittest.TestCase): |
| def test_template_fallback_is_safe_and_evidence_bound(self): |
| selection = SiteSelection( |
| id="S1", |
| selection_type="pin_radius", |
| coordinate_mode="wgs84", |
| geometry_geojson=None, |
| local_geometry=None, |
| anchor_lat=21.0, |
| anchor_lon=70.0, |
| radius_m=250, |
| area_sqm=196000, |
| perimeter_m=1570, |
| centroid=(21.0, 70.0), |
| bbox=None, |
| unit_source="meters", |
| accuracy_label="pin-radius approximation", |
| limitations=["Pin-radius is approximate."], |
| ) |
| evidence = [ |
| make_evidence( |
| category="Climate", |
| finding="Forecast/current climate was retrieved for site-visit timing.", |
| source_name="Open-Meteo", |
| source_url="https://open-meteo.com/", |
| source_type="public weather API", |
| resolution_or_scope="forecast/current", |
| confidence="medium", |
| limitation="Not long-term climate.", |
| design_implication="Use for site-visit timing.", |
| verification_needed="Check weather locally.", |
| output_label="public_data", |
| ) |
| ] |
|
|
| brief = build_assistant_brief( |
| selection=selection, |
| evidence_rows=evidence, |
| warnings=[], |
| project_type="resort", |
| ) |
|
|
| self.assertIn("template fallback active", brief) |
| self.assertRegex(brief, r"E\d{3}: Forecast/current climate") |
| self.assertIn("radius-based context analysis", brief) |
| self.assertEqual(find_forbidden_phrases(brief), []) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
