Add 'python/robomp/' from commit '553fd1cfcf59e4c501c54fc81bc083ffd2ca007b'

git-subtree-dir: python/robomp
git-subtree-mainline: 4f6e70f779
git-subtree-split: 553fd1cfcf

python/robomp/tests/test_worker_pragmas.py

@@ -0,0 +1,80 @@
+"""Worker-side pragma resolution: model + thinking overrides."""
+
+from __future__ import annotations
+
+import pytest
+
+from robomp.config import Settings, reset_settings_cache
+from robomp.worker import DirectiveInfo, _resolve_pragma_overrides
+
+
+@pytest.fixture
+def settings_with_pool(monkeypatch: pytest.MonkeyPatch, env: dict[str, str]) -> Settings:
+    monkeypatch.setenv(
+        "ROBOMP_MODEL",
+        "p-anthropic/claude-sonnet-4-6,p-openai/gpt-5.5,p-openai/gpt-5.5-mini",
+    )
+    reset_settings_cache()
+    return Settings()  # type: ignore[call-arg]
+
+
+def test_no_directive_means_no_override(settings_with_pool: Settings) -> None:
+    assert _resolve_pragma_overrides(None, settings_with_pool) == (None, None)
+
+
+def test_directive_without_pragmas_means_no_override(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run it", author="can1357")
+    assert _resolve_pragma_overrides(directive, settings_with_pool) == (None, None)
+
+
+def test_model_pragma_resolves_to_pool_entry(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "gpt"),))
+    model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override == "p-openai/gpt-5.5"
+    assert thinking_override is None
+
+
+def test_model_alias_exact_short_name(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "gpt-5.5-mini"),))
+    model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override == "p-openai/gpt-5.5-mini"
+
+
+def test_unmatched_model_alias_falls_back_to_random_pick(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run", author="can1357", pragmas=(("model", "qwen"),))
+    model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override is None
+
+
+def test_thinking_pragma_normalized(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run", author="can1357", pragmas=(("thinking", "LOW"),))
+    model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override is None
+    assert thinking_override == "low"
+
+
+def test_unknown_thinking_level_dropped(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(body="run", author="can1357", pragmas=(("thinking", "ultra"),))
+    _, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert thinking_override is None
+
+
+def test_both_pragmas_resolved_together(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(
+        body="run",
+        author="can1357",
+        pragmas=(("model", "claude"), ("thinking", "medium")),
+    )
+    model_override, thinking_override = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override == "p-anthropic/claude-sonnet-4-6"
+    assert thinking_override == "medium"
+
+
+def test_last_value_wins_for_duplicate_keys(settings_with_pool: Settings) -> None:
+    directive = DirectiveInfo(
+        body="run",
+        author="can1357",
+        pragmas=(("model", "claude"), ("model", "gpt")),
+    )
+    model_override, _ = _resolve_pragma_overrides(directive, settings_with_pool)
+    assert model_override == "p-openai/gpt-5.5"