"""Runs the test suite in this workspace against solution.py, records the result, and terminates the session."""


import json
import re
import subprocess
from pathlib import Path


WORKSPACE = Path(__file__).parent
RESULT_PATH = Path("/tmp/results.json")



def main():
    proc = subprocess.run(
        ["python", "-m", "pytest", "test_solution.py", "-v"],
        cwd=WORKSPACE, capture_output=True, text=True,
    )
    out = proc.stdout + proc.stderr
    passed = int(m.group(1)) if (m := re.search(r"(\d+) passed", out)) else 0
    failed = int(m.group(1)) if (m := re.search(r"(\d+) failed", out)) else 0
    total = passed + failed
    reward = passed / total if total else 0.0
    desc = WORKSPACE / "description.md"
    task = desc.read_text().splitlines()[0].lstrip("# ").strip() if desc.exists() else WORKSPACE.name
    record = {
        "task": task,
        "reward": reward,
    }
    RESULT_PATH.write_text(json.dumps(record, indent=2))



if __name__ == "__main__":
    main()

"""Tests for is_prime."""


import pytest


from solution import is_prime as candidate



@pytest.mark.parametrize("n, expected", [
    (0, False),
    (1, False),
    (2, True),
    (7, False),
    (9, False),
    (13, True),
])
def test_is_prime(n, expected):
    assert candidate(n) is expected