Spaces:
Runtime error
Runtime error
| """Unit tests for baseline agents.""" | |
| import pytest | |
| pytest.importorskip("openenv") | |
| from baseline import HeuristicRoutingAgent, RandomRoutingAgent | |
| from server.env import EcoLLMInferenceRoutingEnvironment | |
| from server.models import Strategy | |
| def test_random_agent_creates_valid_actions() -> None: | |
| """Random agent produces valid RLAction values.""" | |
| agent = RandomRoutingAgent() | |
| env = EcoLLMInferenceRoutingEnvironment() | |
| obs = env.reset(task_id="task_1") | |
| action = agent.get_action(obs) | |
| assert action.strategy is not None | |
| assert action.model_choice is not None | |
| assert isinstance(action.exit_flag, bool) | |
| def test_heuristic_agent_prefers_cache() -> None: | |
| """Heuristic agent should prioritize cache hits.""" | |
| agent = HeuristicRoutingAgent() | |
| env = EcoLLMInferenceRoutingEnvironment() | |
| obs = env.reset(task_id="task_1") | |
| obs.cache_contents.append(obs.query) | |
| action = agent.get_action(obs) | |
| assert action.strategy == Strategy.USE_CACHE | |
| def test_random_agent_deterministic_with_seed() -> None: | |
| """Random agent should be deterministic with the same seed.""" | |
| agent1 = RandomRoutingAgent(seed=42) | |
| agent2 = RandomRoutingAgent(seed=42) | |
| env = EcoLLMInferenceRoutingEnvironment() | |
| obs = env.reset(task_id="task_1") | |
| action1 = agent1.get_action(obs) | |
| action2 = agent2.get_action(obs) | |
| assert action1.strategy == action2.strategy | |
| assert action1.model_choice == action2.model_choice | |
| assert action1.exit_flag == action2.exit_flag | |