97ea07c231
P2级改进：基于真实核实的2026官方分数线更新crowd_db 真实数据来源： - 湖南：搜狐教育（本科历史类446/物理类400） - 江苏：微博/教育在线（本科历史类484/物理类456，特控历史类532/物理类513） - 广东：搜狐/新京报（本科历史类440/物理类425，特控历史类546/物理类539） - 山东：高考100（一段441，特控521） - 河北：教育在线（本科历史类485/物理类443，特控历史类542/物理类510） - 河南：微博（本科历史类459/物理类419，特殊类型历史类534/物理类513） 实现内容: 1. 6省data_year: 2025 -> 2026 2. 更新source_url指向官方公布链接 3. 增加quality_note标注2026官方分数线已接入 4. 调整check_crowd_db_consistency.py：允许多年份共存 5. 修复测试以适应过渡期 验证: pytest 155 passed, 3 skipped; consistency check通过
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
"""gaokao-data-trace CLI tests (T3.4)."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from data.crowd_db.cli import main as cli_main
|
||
|
||
PROJECT_ROOT = Path(__file__).resolve().parents[3]
|
||
SCRIPT_PATH = PROJECT_ROOT / "scripts" / "gaokao-data-trace"
|
||
|
||
|
||
def _run_cli(*args: str) -> subprocess.CompletedProcess[str]:
|
||
return subprocess.run(
|
||
[sys.executable, str(SCRIPT_PATH), *args],
|
||
cwd=PROJECT_ROOT,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
|
||
|
||
def test_trace_cli_json_output_contains_matches() -> None:
|
||
result = _run_cli("长沙理工大学")
|
||
assert result.returncode == 0, result.stderr
|
||
|
||
payload = json.loads(result.stdout)
|
||
assert payload["query"] == "长沙理工大学"
|
||
assert payload["match_count"] >= 1
|
||
assert any(match["school"] == "长沙理工大学" for match in payload["matches"])
|
||
|
||
hunan_match = next(
|
||
match for match in payload["matches"] if match["province"] == "湖南"
|
||
)
|
||
assert hunan_match["data_year"] in (2025, 2026) # 过渡期：湖南已切到 2026
|
||
assert hunan_match["source_url"].startswith("https://")
|
||
assert 0 <= hunan_match["confidence"] <= 1
|
||
assert hunan_match["source_type"] == "report"
|
||
assert hunan_match["raw_source_type"] == "manual_summary"
|
||
|
||
|
||
def test_trace_cli_human_output_contains_required_lines(
|
||
capsys: pytest.CaptureFixture[str],
|
||
) -> None:
|
||
exit_code = cli_main(["--human", "长沙理工大学"])
|
||
captured = capsys.readouterr()
|
||
|
||
assert exit_code == 0
|
||
assert "query: 长沙理工大学" in captured.out
|
||
assert "湖南 / 2026年数据 / 长沙理工大学 / 会计学" in captured.out
|
||
assert "source_url: https://" in captured.out
|
||
assert "confidence: 0.85" in captured.out
|
||
assert "quality_level: high (A级（高置信）)" in captured.out
|
||
|
||
|
||
def test_trace_cli_missing_school_returns_nonzero(
|
||
capsys: pytest.CaptureFixture[str],
|
||
) -> None:
|
||
exit_code = cli_main(["不存在的测试院校XYZ"])
|
||
captured = capsys.readouterr()
|
||
|
||
assert exit_code == 1
|
||
assert "不存在的测试院校XYZ" in captured.err
|