wow

ai-engine/tests/test_market_anchor.py

@@ -5,6 +5,9 @@ import sys
 
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
+# tests must be deterministic: never consult the DB source for corrections
+os.environ["MARKET_ANCHOR_DB"] = "0"
+
 from models.market_anchor import devig, home_favorite_delta, apply_home_correction
 
 
@@ -51,6 +54,83 @@ def test_apply_home_correction_keeps_distribution_valid():
     assert _approx(q[0], 0.30)
 
 
+def test_corrections_artifact_loaded_and_fallback():
+    import json
+    import tempfile
+    from models import market_anchor as ma
+
+    # 1) valid artifact -> values come from the file
+    with tempfile.NamedTemporaryFile(
+        "w", suffix=".json", delete=False, encoding="utf-8"
+    ) as fh:
+        json.dump(
+            {"version": "test", "corrections": {"ms_home": [
+                {"lo": 0.60, "hi": 0.70, "delta": 0.042},
+            ]}},
+            fh,
+        )
+        path = fh.name
+    try:
+        os.environ["MARKET_ANCHOR_CORRECTIONS_PATH"] = path
+        ma.reload_corrections()
+        assert _approx(ma.home_favorite_delta(0.65), 0.042)
+        # band not in the artifact -> the STATIC PRIOR applies (silence must
+        # not erase proven knowledge); 0.45-0.55 static prior is 0.010
+        assert _approx(ma.home_favorite_delta(0.50), 0.010)
+
+        # 2) malformed artifact -> static fallback, never crashes
+        with open(path, "w", encoding="utf-8") as fh2:
+            fh2.write("{not json")
+        ma.reload_corrections()
+        assert ma.home_favorite_delta(0.65) > 0.0    # fallback band value
+        assert _approx(ma.home_favorite_delta(0.65), 0.028)
+    finally:
+        os.environ.pop("MARKET_ANCHOR_CORRECTIONS_PATH", None)
+        ma.reload_corrections()
+        os.unlink(path)
+
+
+def test_away_corrections_only_from_artifact():
+    import json
+    import tempfile
+    from models import market_anchor as ma
+
+    # without an artifact: away correction must be ZERO (earned, not assumed).
+    # (Point the env path at a nonexistent file: the repo now SHIPS a fitted
+    # artifact, so "no artifact" must be simulated explicitly.)
+    os.environ["MARKET_ANCHOR_CORRECTIONS_PATH"] = os.path.join(
+        os.path.dirname(__file__), "does_not_exist.json"
+    )
+    ma.reload_corrections()
+    assert ma.away_favorite_delta(0.65) == 0.0
+    base = ma.apply_corrections(0.20, 0.20, 0.60)
+    assert _approx(base[2], 0.60)  # away untouched without artifact
+
+    with tempfile.NamedTemporaryFile(
+        "w", suffix=".json", delete=False, encoding="utf-8"
+    ) as fh:
+        json.dump(
+            {"version": "t2", "corrections": {
+                "ms_home": [{"lo": 0.45, "hi": 0.55, "delta": 0.010}],
+                "ms_away": [{"lo": 0.55, "hi": 0.65, "delta": 0.020}],
+            }},
+            fh,
+        )
+        path = fh.name
+    try:
+        os.environ["MARKET_ANCHOR_CORRECTIONS_PATH"] = path
+        ma.reload_corrections()
+        assert _approx(ma.away_favorite_delta(0.60), 0.020)
+        p1, px, p2 = ma.apply_corrections(0.20, 0.20, 0.60)
+        assert p2 > 0.60                      # away favourite lifted
+        assert _approx(p1 + px + p2, 1.0)     # still a valid distribution
+        assert p1 < 0.20 and px < 0.20        # others renormalised down
+    finally:
+        os.environ.pop("MARKET_ANCHOR_CORRECTIONS_PATH", None)
+        ma.reload_corrections()
+        os.unlink(path)
+
+
 if __name__ == "__main__":
     fns = [v for k, v in sorted(globals().items()) if k.startswith("test_")]
     for fn in fns: