hangshuo652
bc1d56d1a4
P0.6: gcov infrastructure P1: extract_structure output expansion (11 new feature fields) P2: Confusion group rule engine (8 pairs + contradiction + backtrack) P3: 4-factor confidence calculation + quality gate update P4: 33+2 COBOL program type test samples (22 files, 7 categories) P5: parametrized/ test data generation engine P6: japanese_data.py lookup tables P7-10: Type-specific test suites (~159 parametrized tests) P11: Full classification pipeline (classify_program) + orchestrator integration P12: Documentation (module-interfaces, test-plan v3.0, coverage-matrix) Architecture decisions: - classification_pipeline/ merged to hina/pipeline/ - parametrized/ as independent module - japanese_data.py as root-level file - hina/__all__ only exports classify_program() Co-Authored-By: Claude <noreply@anthropic.com>
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
"""OU-01~02: cobol_testgen output 模块 — JSON / 输入文件输出"""
|
|
|
|
import sys, os, json, tempfile
|
|
from pathlib import Path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
|
|
from cobol_testgen.output import output_json, output_input_files
|
|
|
|
|
|
def test_output_json_basic():
|
|
"""OU-01: 3条记录 → 有效 JSON"""
|
|
records = [{"WS-A": "1", "WS-B": "2"}, {"WS-A": "3", "WS-B": "4"}]
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
outpath = Path(tmp) / "output.json"
|
|
output_json(records, outpath)
|
|
assert outpath.exists()
|
|
data = json.loads(outpath.read_text(encoding="utf-8"))
|
|
assert len(data) == 2
|
|
assert data[0]["WS-A"] == "1"
|
|
|
|
def test_output_json_with_roles():
|
|
"""带角色分组的 JSON 输出"""
|
|
records = [{"WS-A": "1", "WS-B": "2"}]
|
|
roles = {"WS-A": "input", "WS-B": "output"}
|
|
fd_fields = {"FILE1": {"WS-A"}}
|
|
field_to_fd = {"WS-A": "FILE1"}
|
|
open_dir = {"FILE1": "INPUT"}
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
outpath = Path(tmp) / "output.json"
|
|
output_json(records, outpath, roles, fd_fields, field_to_fd, open_dir)
|
|
assert outpath.exists()
|
|
|
|
def test_output_json_empty():
|
|
"""空记录 → 空数组"""
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
outpath = Path(tmp) / "empty.json"
|
|
output_json([], outpath)
|
|
assert json.loads(outpath.read_text(encoding="utf-8")) == []
|
|
|
|
def test_output_input_files_basic():
|
|
"""OU-02: 输入文件输出"""
|
|
records = [{"WS-A": "1"}]
|
|
roles = {"WS-A": "input"}
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
output_input_files(records, tmp, "TESTPGM", roles, {}, {}, {})
|
|
assert os.path.isdir(tmp)
|