fix: refill from expanded candidate history
This commit is contained in:
34
tests/test_agent_entrypoint.py
Normal file
34
tests/test_agent_entrypoint.py
Normal file
@@ -0,0 +1,34 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import importlib.util
|
||||
import os
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch
|
||||
|
||||
|
||||
ROOT_DIR = Path(__file__).resolve().parents[1]
|
||||
SPEC = importlib.util.spec_from_file_location("modelhub_agent_entrypoint", ROOT_DIR / "main.py")
|
||||
if SPEC is None or SPEC.loader is None:
|
||||
raise RuntimeError("Unable to load the hosted agent entrypoint")
|
||||
ENTRYPOINT = importlib.util.module_from_spec(SPEC)
|
||||
SPEC.loader.exec_module(ENTRYPOINT)
|
||||
|
||||
|
||||
class HostedAgentEntrypointTests(unittest.TestCase):
|
||||
def test_hosted_worker_always_uses_unlimited_cycle_submissions(self) -> None:
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"MODELHUB_AGENT_MAX_SUBMITS_PER_RUN": "5",
|
||||
"MODELHUB_AGENT_EXTRA_ARGS": "--max-submits-per-run 3",
|
||||
},
|
||||
clear=False,
|
||||
):
|
||||
command = ENTRYPOINT._worker_command()
|
||||
|
||||
self.assertEqual(["--max-submits-per-run", "0"], command[-2:])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user