| |
| """ |
| CropAI Automated Test Suite for Advanced Innovation Engines: |
| 1. Vernacular Voice AI Copilot (Chhattisgarhi, Rural Hindi, Bhojpuri) |
| 2. Soil Health Card AI Digitizer & Stage-Specific NPK Fertilizer Optimizer |
| """ |
|
|
| import unittest |
| from fastapi.testclient import TestClient |
|
|
| from backend.main import app |
| from backend.voice_copilot import generate_vernacular_advisory, normalize_vernacular_text, SUPPORTED_DIALECTS |
| from backend.soil_health import digitize_and_optimize_fertilizer |
|
|
|
|
| class TestAdvancedInnovations(unittest.TestCase): |
| @classmethod |
| def setUpClass(cls): |
| cls.client = TestClient(app) |
|
|
| |
| |
| |
| def test_voice_supported_dialects_endpoint(self): |
| """Verify GET /api/v1/voice/supported-dialects returns Chhattisgarhi and Hindi.""" |
| resp = self.client.get("/api/v1/voice/supported-dialects") |
| self.assertEqual(resp.status_code, 200) |
| data = resp.json() |
| self.assertEqual(data["status"], "success") |
| self.assertGreaterEqual(data["total_dialects"], 3) |
| codes = [d["code"] for d in data["dialects"]] |
| self.assertIn("ch-IN", codes) |
| self.assertIn("hi-IN", codes) |
| self.assertIn("bho-IN", codes) |
|
|
| def test_chhattisgarhi_vernacular_normalization(self): |
| """Verify normalization of Chhattisgarhi words like 'करपा' and 'माहू'.""" |
| norm = normalize_vernacular_text("हमार धान म करपा लग गे हे", dialect="ch-IN") |
| self.assertEqual(norm["standardized_crop"], "धान (Rice)") |
| self.assertIn("Blast", norm["standardized_pathology"]) |
|
|
| def test_voice_advisory_endpoint(self): |
| """Verify POST /api/v1/voice/transcribe-and-advise returns dialect speech and tank dosage.""" |
| payload = { |
| "user_query": "हमार धान म करपा धर ले हे, का दवाई छिड़कन?", |
| "dialect": "ch-IN", |
| "crop": "धान (Rice)" |
| } |
| resp = self.client.post("/api/v1/voice/transcribe-and-advise", json=payload) |
| self.assertEqual(resp.status_code, 200) |
| data = resp.json() |
| self.assertEqual(data["status"], "success") |
| self.assertEqual(data["dialect"], "ch-IN") |
| self.assertIn("जय जोहार", data["advisory_text"]) |
| self.assertIn("16 लीटर", data["advisory_text"]) |
| self.assertIn("Tricyclazole", data["tank_prescription"]["chemical_dose"]) |
|
|
| |
| |
| |
| def test_soil_health_digitization_and_splits(self): |
| """Verify 3-stage commercial fertilizer split calculations adhering to Rule 9.""" |
| params = { |
| "ph": 6.8, |
| "organic_carbon_pct": 0.50, |
| "available_n_kg_ha": 210.0, |
| "available_p_kg_ha": 12.0, |
| "available_k_kg_ha": 160.0, |
| "zinc_ppm": 0.45 |
| } |
| result = digitize_and_optimize_fertilizer(soil_params=params, crop="Rice", area_acres=1.0) |
| self.assertEqual(result["status"], "success") |
| self.assertGreater(result["total_commercial_fertilizer"]["dap_kg"], 10.0) |
| self.assertGreater(result["total_commercial_fertilizer"]["urea_kg"], 15.0) |
| self.assertEqual(len(result["split_schedule"]), 3) |
| self.assertIn("रोपाई", result["split_schedule"][0]["stage"]) |
|
|
| def test_soil_health_active_blast_warning(self): |
| """Verify Rule 9 safety check: stops urea application if crop has active blast.""" |
| result = digitize_and_optimize_fertilizer( |
| soil_params={"ph": 6.8}, |
| crop="Rice", |
| area_acres=1.0, |
| has_active_blast=True |
| ) |
| self.assertIsNotNone(result["safety_alert"]) |
| self.assertIn("यूरिया का छिड़काव तुरंत रोक दें", result["safety_alert"]) |
|
|
| def test_soil_health_api_endpoint(self): |
| """Verify POST /api/v1/soil/analyze-card returns bags and stage schedule.""" |
| payload = { |
| "crop": "Rice", |
| "area_acres": 2.0, |
| "ph": 6.5, |
| "available_n_kg_ha": 200.0, |
| "has_active_blast": False |
| } |
| resp = self.client.post("/api/v1/soil/analyze-card", json=payload) |
| self.assertEqual(resp.status_code, 200) |
| data = resp.json() |
| self.assertEqual(data["status"], "success") |
| self.assertGreater(data["total_commercial_fertilizer"]["urea_bags_45kg"], 0.5) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
