holy agent

tests/test_enrichment.py

@@ -0,0 +1,46 @@
+import unittest
+
+from hpcsim.enrichment import Enricher
+
+
+class TestEnrichment(unittest.TestCase):
+    def test_basic_ranges(self):
+        e = Enricher()
+        sample = {
+            "lap": 1,
+            "speed": 250,
+            "throttle": 0.8,
+            "brake": 0.1,
+            "tire_compound": "medium",
+            "fuel_level": 0.6,
+            "ers": 0.5,
+            "track_temp": 35,
+            "rain_probability": 0.1,
+        }
+        out = e.enrich(sample)
+        self.assertIn("aero_efficiency", out)
+        self.assertTrue(0.0 <= out["aero_efficiency"] <= 1.0)
+        self.assertTrue(0.0 <= out["tire_degradation_index"] <= 1.0)
+        self.assertTrue(0.0 <= out["ers_charge"] <= 1.0)
+        self.assertTrue(0.0 <= out["fuel_optimization_score"] <= 1.0)
+        self.assertTrue(0.0 <= out["driver_consistency"] <= 1.0)
+        self.assertIn(out["weather_impact"], {"low", "medium", "high"})
+
+    def test_stateful_wear_increases(self):
+        e = Enricher()
+        prev = 0.0
+        for lap in range(1, 6):
+            out = e.enrich({
+                "lap": lap,
+                "speed": 260,
+                "throttle": 0.9,
+                "brake": 0.05,
+                "tire_compound": "soft",
+                "fuel_level": 0.7,
+            })
+            self.assertGreaterEqual(out["tire_degradation_index"], prev)
+            prev = out["tire_degradation_index"]
+
+
+if __name__ == "__main__":
+    unittest.main()