Files
gaokao-volunteer-system/tests/test_audit_engine_major_validation_phase2.py
Hermes Agent edc5b11230
Some checks failed
CI / pytest (Python 3.10) (push) Has been cancelled
CI / pytest (Python 3.11) (push) Has been cancelled
CI / pytest (Python 3.12) (push) Has been cancelled
feat(audit): validate majors in structured audit flow
2026-06-17 16:02:31 +08:00

123 lines
4.3 KiB
Python

from __future__ import annotations
import json
from pathlib import Path
from typing import Any, cast
from data.majors_catalog.loader import MajorsCatalogLoader
from data.rules.audit_engine import AuditEngine
from data.rules.loader import RuleLoader
def _write_truth_and_catalog(root: Path) -> tuple[Path, Path, Path]:
truth_root = root / "truth"
province_dir = truth_root / "province"
province_dir.mkdir(parents=True, exist_ok=True)
(truth_root / "national.yaml").write_text(
"scope: national\nyear: 2026\nversion: '2026.1'\nrules: {}\n",
encoding="utf-8",
)
(province_dir / "hunan.yaml").write_text(
"scope: province\nprovince: 湖南\nyear: 2026\nversion: '2026.1'\nstatus: active\nrules:\n max_volunteers:\n title: 志愿上限\n severity: fatal\n value:\n max_volunteers: 45\n source_evidence_id: hunan-2026-max-volunteers\n effective_date: '2026-01-01'\n status: active\n",
encoding="utf-8",
)
catalog_root = root / "catalog"
national = catalog_root / "national"
national.mkdir(parents=True, exist_ok=True)
payload = {
"year": 2024,
"version": "2024.1",
"coverage_mode": "mvp_subset",
"source": "教育部普通高等学校本科专业目录(人工摘录校对)",
"source_url": "https://www.moe.gov.cn/",
"last_verified_at": "2026-06-17",
"majors": [
{
"code": "080901",
"name": "计算机科学与技术",
"discipline": "工学",
"category": "计算机类",
"degree": "工学学士",
"is_directional": False,
"status": "active",
"year_added": 1998,
"year_removed": None,
"notes": None,
},
{
"code": "120201K",
"name": "工商管理",
"discipline": "管理学",
"category": "工商管理类",
"degree": "管理学学士",
"is_directional": True,
"status": "deprecated",
"year_added": 1998,
"year_removed": 2024,
"notes": "示例停用",
},
],
}
for name in ("2024.json", "latest.json"):
(national / name).write_text(
json.dumps(payload, ensure_ascii=False, indent=2), encoding="utf-8"
)
plan_path = root / "plan.json"
plan_path.write_text(
json.dumps(
{
"province": "湖南",
"items": [
{
"school_name": "北京大学",
"major_names": ["计算机科学与技术", "不存在专业", "工商管理"],
}
],
},
ensure_ascii=False,
indent=2,
),
encoding="utf-8",
)
return truth_root, catalog_root, plan_path
def test_audit_engine_major_validation_marks_missing_and_non_active_majors(
tmp_path: Path,
) -> None:
truth_root, catalog_root, plan_path = _write_truth_and_catalog(tmp_path)
loader = RuleLoader.from_truth_root(truth_root)
catalog = MajorsCatalogLoader.from_catalog_root(catalog_root)
engine = AuditEngine(loader, majors_loader=catalog)
plan = json.loads(plan_path.read_text(encoding="utf-8"))
result = engine.audit_plan("湖南", plan)
issues = cast(list[dict[str, Any]], result["issues"])
rule_ids = {issue["rule_id"] for issue in issues}
assert "MAJORS.not_found" in rule_ids
assert "MAJORS.non_active" in rule_ids
assert result["overall_pass"] is False
def test_audit_engine_major_validation_passes_when_all_majors_are_active(
tmp_path: Path,
) -> None:
truth_root, catalog_root, _ = _write_truth_and_catalog(tmp_path)
loader = RuleLoader.from_truth_root(truth_root)
catalog = MajorsCatalogLoader.from_catalog_root(catalog_root)
engine = AuditEngine(loader, majors_loader=catalog)
result = engine.audit_plan(
"湖南",
{
"province": "湖南",
"items": [{"school_name": "北京大学", "major_names": ["计算机科学与技术"]}],
},
)
assert result["issues"] == []
assert result["overall_pass"] is True