| |
| """ |
| Test Suite for AgriMemory™ Outbreak Recurrence & Healthy Crop Guardrails. |
| Verifies: |
| 1. Healthy crops (Tomato___healthy, Rice___healthy) are NEVER recorded or flagged as at-risk outbreaks. |
| 2. check_seasonal_disease_alerts handles historical dates cleanly without NameError. |
| 3. Prophylactic care plans are generated only for valid disease IDs. |
| """ |
|
|
| import unittest |
| import os |
| import tempfile |
| from datetime import datetime, timedelta |
| from backend.farm_memory import AgriMemoryEngine |
|
|
|
|
| class TestFarmMemoryEngine(unittest.TestCase): |
| def setUp(self): |
| |
| self.temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".json") |
| self.temp_file.close() |
| self.engine = AgriMemoryEngine(storage_path=self.temp_file.name) |
|
|
| def tearDown(self): |
| if os.path.exists(self.temp_file.name): |
| os.remove(self.temp_file.name) |
|
|
| def test_healthy_crop_flagged_correctly(self): |
| """Verify that healthy crop diagnoses are marked with is_healthy=True.""" |
| res = self.engine.save_diagnosis_record( |
| crop="Tomato", |
| disease="Healthy Foliage (स्वस्थ पौधा)", |
| disease_id="Tomato___healthy", |
| severity="Healthy", |
| confidence=0.98, |
| temperature_c=25.0, |
| humidity_pct=70.0, |
| weather_condition="Clear Sky", |
| diagnosis_date=datetime.now().isoformat() |
| ) |
| self.assertEqual(res["status"], "success") |
| self.assertTrue(res["record"]["is_healthy"]) |
|
|
| def test_healthy_crop_never_triggers_seasonal_alarm(self): |
| """Verify that past healthy records within current calendar window NEVER trigger risk alerts.""" |
| |
| last_year_date = (datetime.now() - timedelta(days=365)).isoformat() |
| self.engine.save_diagnosis_record( |
| crop="Tomato", |
| disease="Healthy Tomato Leaf", |
| disease_id="Tomato___healthy", |
| severity="Healthy", |
| confidence=0.99, |
| temperature_c=26.0, |
| humidity_pct=75.0, |
| weather_condition="Optimal", |
| diagnosis_date=last_year_date |
| ) |
|
|
| |
| alerts = self.engine.check_seasonal_disease_alerts( |
| current_date=datetime.now().isoformat(), |
| current_temp_c=26.0, |
| current_humidity_pct=75.0 |
| ) |
|
|
| |
| for alert in alerts: |
| self.assertNotEqual(alert.get("disease_id"), "Tomato___healthy") |
| self.assertFalse("healthy" in alert.get("disease", "").lower()) |
| self.assertFalse("स्वस्थ" in alert.get("disease", "")) |
|
|
| def test_real_disease_triggers_prior_year_alarm(self): |
| """Verify that real severe outbreaks from prior year correctly trigger seasonal warnings without NameError.""" |
| last_year_date = (datetime.now() - timedelta(days=365)).isoformat() |
| self.engine.save_diagnosis_record( |
| crop="Wheat", |
| disease="Yellow Rust (गेहूं का पीला रतुआ)", |
| disease_id="Wheat___Yellow_Rust", |
| severity="Severe", |
| confidence=0.95, |
| temperature_c=27.0, |
| humidity_pct=88.0, |
| weather_condition="Optimal Humidity", |
| diagnosis_date=last_year_date |
| ) |
|
|
| alerts = self.engine.check_seasonal_disease_alerts( |
| current_date=datetime.now().isoformat(), |
| current_temp_c=27.0, |
| current_humidity_pct=88.0 |
| ) |
|
|
| |
| rust_alerts = [a for a in alerts if "Wheat___Yellow_Rust" in a.get("disease_id", "")] |
| self.assertTrue(len(rust_alerts) >= 1) |
| self.assertTrue(rust_alerts[0]["recurrence_risk_score"] > 80.0) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
