Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/hermes_cli/test_model_data_policy_guard.py

@@ -0,0 +1,39 @@
+"""Tests for the data-training-tier selection guard."""
+
+from hermes_cli.model_data_policy_guard import (
+    DataTrainingWarning,
+    data_training_warning,
+)
+
+
+def test_fires_on_meta_contributor():
+    w = data_training_warning("muse-spark-1.2-contributor", provider="meta-ai")
+    assert isinstance(w, DataTrainingWarning)
+    assert w.model == "muse-spark-1.2-contributor"
+    assert "train" in w.message.lower()
+    assert "-contributor" in w.message.lower() or "contributor" in w.message.lower()  # mentions the tier
+    # Points to Meta's live pricing page instead of hardcoding prices.
+    assert "pricing and rate limits" in w.message.lower()
+    assert "$0.10" not in w.message and "$0.20" not in w.message and "$0.002" not in w.message
+    assert "prompts and completions" in w.message.lower()
+    assert "dev.meta.ai/docs/pricing-rate-limits" in w.message
+
+
+def test_silent_on_non_contributor_muse():
+    assert data_training_warning("muse-spark-1.2", provider="meta-ai") is None
+    assert data_training_warning("muse-spark-1.1", provider="meta-ai") is None
+
+
+def test_silent_on_unrelated_models():
+    for m in ("anthropic/claude-opus-4.8", "gpt-5.6-sol", "deepseek-v4-pro", ""):
+        assert data_training_warning(m, provider="anthropic") is None
+
+
+def test_fires_regardless_of_provider_string():
+    # id-keyed: must fire even if selected via custom/gateway (no meta-ai provider)
+    assert data_training_warning("muse-spark-1.2-contributor", provider="custom") is not None
+    assert data_training_warning("muse-spark-1.2-contributor") is not None
+
+
+def test_case_insensitive():
+    assert data_training_warning("MUSE-SPARK-1.2-CONTRIBUTOR", provider="meta-ai") is not None