maris-ai-master / core-python /tests /test_coder_execution_benchmark_dataset.py
MarisUK's picture
Maris AI model sync
f440f03 verified
"""Tests for coder execution benchmark dataset coverage."""
import json
from pathlib import Path
from maris_core.text.benchmark import load_chat_benchmark_dataset
def test_coder_execution_benchmark_dataset_covers_primary_languages() -> None:
dataset_path = Path(
"/home/runner/work/Maris-MI/Maris-MI/core-python/evals/coder_execution_benchmark.json"
)
cases = load_chat_benchmark_dataset(dataset_path)
languages = {case.execution_language for case in cases if case.execution_language}
assert {"python", "typescript", "rust", "sql"}.issubset(languages)
assert all(case.level == "ci" for case in cases)
def test_coder_release_benchmark_dataset_is_execution_and_repo_heavy() -> None:
dataset_path = Path(
"/home/runner/work/Maris-MI/Maris-MI/core-python/evals/coder_release_benchmark.json"
)
cases = load_chat_benchmark_dataset(dataset_path)
execution_cases = [case for case in cases if case.execution_language]
repo_cases = [case for case in cases if "repo-level" in case.tags or "diff" in case.tags]
assert len(cases) >= 10
assert len(execution_cases) >= 5
assert repo_cases
assert any("multi-file" in case.tags for case in cases)
assert any("regression-risk" in case.tags for case in cases)
assert any("incident-recovery" in case.tags for case in cases)
assert any("debugging" in case.tags for case in cases)
def test_coder_sample_dataset_covers_repo_diff_and_incident_work() -> None:
dataset_path = Path("/home/runner/work/Maris-MI/Maris-MI/data/code/sample.jsonl")
records = [
json.loads(line)
for line in dataset_path.read_text(encoding="utf-8").splitlines()
if line.strip()
]
task_types = {str(record.get("metadata", {}).get("task", "")).strip() for record in records}
repo_context_counts = [len(record.get("repo_context", [])) for record in records]
prompts = [str(record.get("prompt", "")) for record in records]
assert {"repo-level", "bugfix", "refactor", "ci-orchestration"}.issubset(task_types)
assert max(repo_context_counts) >= 2
assert any("incident" in prompt.lower() for prompt in prompts)
assert any(str(record.get("diff", "")).strip() for record in records)