feat(harness): RunConfig frozen dataclass + 四层校验 + YAML/CLI/.env 三层加载
- RunConfig: 46 字段 frozen dataclass，从 TRM4 core/harness/config.py 迁移 - 四层校验链：_validate → _validate_edit_budget + _validate_minibatch + _validate_gate - 新增 .env 覆盖层：工程配置（workspace_dir, store_dir）可通过 HARNESS_* 环境变量注入 - 合并优先级：CLI > .env > YAML（CLAUDE.md §4.5） - load_config 支持嵌套 harness 段和扁平 YAML 两种格式 - run_id 改为默认空字符串（CLI-only 字段，YAML 不提供） - resume/fresh 互斥校验不在 config 层（移至 runner.py） - 70 个单元测试全部通过 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,540 @@
|
||||
"""app/harness/config.py 单元测试。
|
||||
|
||||
覆盖 RunConfig 构造、四层校验（_validate → 三个子校验）、
|
||||
YAML + CLI + .env 三层加载优先级。
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
import yaml
|
||||
|
||||
from app.harness.config import RunConfig, _validate, load_config
|
||||
|
||||
# ────────────────────────────── 测试数据工厂 ──────────────────────────────
|
||||
|
||||
|
||||
def _valid_kwargs() -> dict:
|
||||
"""构造一组完整合法的 RunConfig 字段值（使用真实 default.yaml 数据）。"""
|
||||
return {
|
||||
"workspace_dir": Path("workspaces/default"),
|
||||
"store_dir": Path("store"),
|
||||
"mode": "infer",
|
||||
"run_id": "",
|
||||
"concurrency": 12,
|
||||
"max_steps": 15,
|
||||
"skill_mode": "auto",
|
||||
"n_samples": 0,
|
||||
"questions": "benchmarks/Video-MME",
|
||||
"skills_version": "v1",
|
||||
"prompts_version": "v1",
|
||||
"epochs": 1,
|
||||
"diag_size": 200,
|
||||
"diag_correct_ratio": 0.5,
|
||||
"val_size": 30,
|
||||
"val_correct_ratio": 0.5,
|
||||
"edit_budget_start": 5,
|
||||
"edit_budget_end": 2,
|
||||
"batch_size": 15,
|
||||
"min_class_per_batch": 2,
|
||||
"eval_min_per_class": 2,
|
||||
"early_stop_patience": 8,
|
||||
"test_size": 60,
|
||||
"use_slow_momentum": True,
|
||||
"gate_e_confirm": 20.0,
|
||||
"gate_e_provisional": 3.0,
|
||||
"gate_w_net_min": 2,
|
||||
"gate_delta_min": 0.02,
|
||||
"gate_lambda_dir": -0.642,
|
||||
"gate_e_rollback": 10.0,
|
||||
"gate_block": 8,
|
||||
"gate_n_max": 40,
|
||||
"gate_p_low": 0.05,
|
||||
"gate_p_high": 0.95,
|
||||
"gate_probe_quota": 0.2,
|
||||
"gate_gamma_decay": 0.9,
|
||||
"gate_cooldown_steps": 2,
|
||||
"gate_guard_err": 0.10,
|
||||
"skill_update_mode": "patch",
|
||||
"appendix_consolidate_threshold": 6,
|
||||
"batch_correct_ratio": 0.5,
|
||||
"momentum_samples": 20,
|
||||
}
|
||||
|
||||
|
||||
def _make_config(**overrides: object) -> RunConfig:
|
||||
"""用 _valid_kwargs 构造 RunConfig，支持字段覆盖。"""
|
||||
kwargs = _valid_kwargs()
|
||||
kwargs.update(overrides)
|
||||
return RunConfig(**kwargs)
|
||||
|
||||
|
||||
def _yaml_harness_dict() -> dict:
|
||||
"""构造可序列化为 YAML 的合法 harness 配置字典（路径用字符串）。"""
|
||||
d = _valid_kwargs()
|
||||
d["workspace_dir"] = str(d["workspace_dir"])
|
||||
d["store_dir"] = str(d["store_dir"])
|
||||
return d
|
||||
|
||||
|
||||
def _write_yaml(tmp_path: Path, harness_data: dict) -> Path:
|
||||
"""将 harness 配置写入临时 YAML 文件，返回文件路径。"""
|
||||
yaml_path = tmp_path / "experiment.yaml"
|
||||
with open(yaml_path, "w", encoding="utf-8") as f:
|
||||
yaml.dump({"harness": harness_data}, f)
|
||||
return yaml_path
|
||||
|
||||
|
||||
# ──────────────────────────── test_valid_config ────────────────────────────
|
||||
|
||||
|
||||
class TestValidConfig:
|
||||
"""合法参数应能正常构造并通过校验。"""
|
||||
|
||||
def test_default_yaml_values_pass_validation(self) -> None:
|
||||
"""使用 default.yaml 真实默认值构造的 RunConfig 应通过全部校验。"""
|
||||
cfg = _make_config()
|
||||
_validate(cfg)
|
||||
assert cfg.mode == "infer"
|
||||
assert cfg.workspace_dir == Path("workspaces/default")
|
||||
assert cfg.store_dir == Path("store")
|
||||
|
||||
def test_frozen_immutability(self) -> None:
|
||||
"""frozen=True 应禁止字段赋值。"""
|
||||
cfg = _make_config()
|
||||
with pytest.raises(AttributeError):
|
||||
cfg.mode = "train" # type: ignore[misc]
|
||||
|
||||
def test_default_field_values(self) -> None:
|
||||
"""有默认值的字段不传时应使用默认值。"""
|
||||
kwargs = _valid_kwargs()
|
||||
# 不传 run_id / seed / version / resume / fresh，使用默认值
|
||||
kwargs.pop("run_id", None)
|
||||
cfg = RunConfig(**kwargs)
|
||||
assert cfg.run_id == ""
|
||||
assert cfg.seed == "initial"
|
||||
assert cfg.version == ""
|
||||
assert cfg.resume is False
|
||||
assert cfg.fresh is False
|
||||
|
||||
|
||||
# ──────────────────────────── test_mode_validation ────────────────────────
|
||||
|
||||
|
||||
class TestModeValidation:
|
||||
"""mode 字段校验。"""
|
||||
|
||||
@pytest.mark.parametrize("bad_mode", ["unknown", "test", "", "INFER", "Train"])
|
||||
def test_invalid_mode_rejected(self, bad_mode: str) -> None:
|
||||
"""非法 mode 应抛出 ValueError。"""
|
||||
cfg = _make_config(mode=bad_mode)
|
||||
with pytest.raises(ValueError, match="mode"):
|
||||
_validate(cfg)
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"valid_mode", ["infer", "train", "diagnose", "evolve", "eval", "promote"]
|
||||
)
|
||||
def test_all_valid_modes_accepted(self, valid_mode: str) -> None:
|
||||
"""全部合法 mode 应通过校验（diagnose/evolve 需 run_id）。"""
|
||||
overrides: dict = {"mode": valid_mode}
|
||||
if valid_mode in ("diagnose", "evolve"):
|
||||
overrides["run_id"] = "run-001"
|
||||
if valid_mode in ("eval", "promote"):
|
||||
overrides["version"] = "v1"
|
||||
if valid_mode == "promote":
|
||||
overrides["run_id"] = "run-001"
|
||||
cfg = _make_config(**overrides)
|
||||
_validate(cfg)
|
||||
|
||||
def test_diagnose_requires_run_id(self) -> None:
|
||||
"""diagnose 模式缺少 run_id 应报错。"""
|
||||
cfg = _make_config(mode="diagnose", run_id="")
|
||||
with pytest.raises(ValueError, match="run_id"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_evolve_requires_run_id(self) -> None:
|
||||
"""evolve 模式缺少 run_id 应报错。"""
|
||||
cfg = _make_config(mode="evolve", run_id="")
|
||||
with pytest.raises(ValueError, match="run_id"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_eval_requires_version(self) -> None:
|
||||
"""eval 模式缺少 version 应报错。"""
|
||||
cfg = _make_config(mode="eval", version="")
|
||||
with pytest.raises(ValueError, match="version"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_promote_requires_run_id_and_version(self) -> None:
|
||||
"""promote 模式需同时提供 run_id 和 version。"""
|
||||
cfg = _make_config(mode="promote", run_id="", version="v1")
|
||||
with pytest.raises(ValueError, match="run.id"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_concurrency_positive(self) -> None:
|
||||
"""concurrency <= 0 应报错。"""
|
||||
cfg = _make_config(concurrency=0)
|
||||
with pytest.raises(ValueError, match="concurrency"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_max_steps_positive(self) -> None:
|
||||
"""max_steps <= 0 应报错。"""
|
||||
cfg = _make_config(max_steps=-1)
|
||||
with pytest.raises(ValueError, match="max_steps"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_n_samples_non_negative(self) -> None:
|
||||
"""n_samples < 0 应报错。"""
|
||||
cfg = _make_config(n_samples=-1)
|
||||
with pytest.raises(ValueError, match="n_samples"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_epochs_positive(self) -> None:
|
||||
"""epochs <= 0 应报错。"""
|
||||
cfg = _make_config(epochs=0)
|
||||
with pytest.raises(ValueError, match="epochs"):
|
||||
_validate(cfg)
|
||||
|
||||
@pytest.mark.parametrize("bad_skill_mode", ["Auto", "disabled", ""])
|
||||
def test_invalid_skill_mode(self, bad_skill_mode: str) -> None:
|
||||
"""非法 skill_mode 应报错。"""
|
||||
cfg = _make_config(skill_mode=bad_skill_mode)
|
||||
with pytest.raises(ValueError, match="skill_mode"):
|
||||
_validate(cfg)
|
||||
|
||||
@pytest.mark.parametrize("bad_update_mode", ["append", "delete", ""])
|
||||
def test_invalid_skill_update_mode(self, bad_update_mode: str) -> None:
|
||||
"""非法 skill_update_mode 应报错。"""
|
||||
cfg = _make_config(skill_update_mode=bad_update_mode)
|
||||
with pytest.raises(ValueError, match="skill_update_mode"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_appendix_consolidate_threshold_positive(self) -> None:
|
||||
"""appendix_consolidate_threshold < 1 应报错。"""
|
||||
cfg = _make_config(appendix_consolidate_threshold=0)
|
||||
with pytest.raises(ValueError, match="appendix_consolidate_threshold"):
|
||||
_validate(cfg)
|
||||
|
||||
|
||||
# ────────────────────── test_edit_budget_validation ───────────────────────
|
||||
|
||||
|
||||
class TestEditBudgetValidation:
|
||||
"""编辑预算退火校验（_validate_edit_budget）。"""
|
||||
|
||||
def test_start_less_than_end_rejected(self) -> None:
|
||||
"""edit_budget_start < edit_budget_end 应抛出 ValueError。"""
|
||||
cfg = _make_config(edit_budget_start=1, edit_budget_end=5)
|
||||
with pytest.raises(ValueError, match="edit_budget_start"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_end_zero_rejected(self) -> None:
|
||||
"""edit_budget_end <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(edit_budget_start=1, edit_budget_end=0)
|
||||
with pytest.raises(ValueError, match="edit_budget_end"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_equal_values_accepted(self) -> None:
|
||||
"""edit_budget_start == edit_budget_end 应通过。"""
|
||||
cfg = _make_config(edit_budget_start=3, edit_budget_end=3)
|
||||
_validate(cfg)
|
||||
|
||||
|
||||
# ─────────────────────── test_minibatch_validation ───────────────────────
|
||||
|
||||
|
||||
class TestMinibatchValidation:
|
||||
"""mini-batch 自进化闭环参数校验（_validate_minibatch）。"""
|
||||
|
||||
def test_batch_size_zero_rejected(self) -> None:
|
||||
"""batch_size <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(batch_size=0)
|
||||
with pytest.raises(ValueError, match="batch_size"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_min_class_per_batch_equals_batch_size_rejected(self) -> None:
|
||||
"""min_class_per_batch >= batch_size 应抛出 ValueError。"""
|
||||
cfg = _make_config(batch_size=5, min_class_per_batch=5)
|
||||
with pytest.raises(ValueError, match="min_class_per_batch"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_min_class_per_batch_zero_rejected(self) -> None:
|
||||
"""min_class_per_batch < 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(min_class_per_batch=0)
|
||||
with pytest.raises(ValueError, match="min_class_per_batch"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_eval_min_per_class_zero_rejected(self) -> None:
|
||||
"""eval_min_per_class < 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(eval_min_per_class=0)
|
||||
with pytest.raises(ValueError, match="eval_min_per_class"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_early_stop_patience_zero_rejected(self) -> None:
|
||||
"""early_stop_patience <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(early_stop_patience=0)
|
||||
with pytest.raises(ValueError, match="early_stop_patience"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_test_size_zero_rejected(self) -> None:
|
||||
"""test_size <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(test_size=0)
|
||||
with pytest.raises(ValueError, match="test_size"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_batch_correct_ratio_one_rejected(self) -> None:
|
||||
"""batch_correct_ratio >= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(batch_correct_ratio=1.0)
|
||||
with pytest.raises(ValueError, match="batch_correct_ratio"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_batch_correct_ratio_negative_rejected(self) -> None:
|
||||
"""batch_correct_ratio < 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(batch_correct_ratio=-0.1)
|
||||
with pytest.raises(ValueError, match="batch_correct_ratio"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_momentum_samples_zero_rejected(self) -> None:
|
||||
"""momentum_samples < 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(momentum_samples=0)
|
||||
with pytest.raises(ValueError, match="momentum_samples"):
|
||||
_validate(cfg)
|
||||
|
||||
|
||||
# ─────────────────────────── test_gate_validation ────────────────────────
|
||||
|
||||
|
||||
class TestGateValidation:
|
||||
"""CE-Gate 参数校验（_validate_gate）。"""
|
||||
|
||||
def test_e_confirm_at_one_rejected(self) -> None:
|
||||
"""gate_e_confirm <= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_e_confirm=1.0, gate_e_provisional=1.0)
|
||||
with pytest.raises(ValueError, match="gate_e_confirm"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_e_provisional_exceeds_confirm_rejected(self) -> None:
|
||||
"""gate_e_provisional > gate_e_confirm 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_e_confirm=10.0, gate_e_provisional=15.0)
|
||||
with pytest.raises(ValueError, match="gate_e_provisional"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_e_provisional_at_one_rejected(self) -> None:
|
||||
"""gate_e_provisional <= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_e_provisional=0.5)
|
||||
with pytest.raises(ValueError, match="gate_e_provisional"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_e_rollback_at_one_rejected(self) -> None:
|
||||
"""gate_e_rollback <= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_e_rollback=1.0)
|
||||
with pytest.raises(ValueError, match="gate_e_rollback"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_w_net_min_zero_rejected(self) -> None:
|
||||
"""gate_w_net_min < 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_w_net_min=0)
|
||||
with pytest.raises(ValueError, match="gate_w_net_min"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_lambda_dir_positive_rejected(self) -> None:
|
||||
"""gate_lambda_dir >= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_lambda_dir=0.5)
|
||||
with pytest.raises(ValueError, match="gate_lambda_dir"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_lambda_dir_zero_rejected(self) -> None:
|
||||
"""gate_lambda_dir == 0 也应报错。"""
|
||||
cfg = _make_config(gate_lambda_dir=0.0)
|
||||
with pytest.raises(ValueError, match="gate_lambda_dir"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_block_exceeds_n_max_rejected(self) -> None:
|
||||
"""gate_block > gate_n_max 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_block=50, gate_n_max=40)
|
||||
with pytest.raises(ValueError, match="gate_block"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_block_zero_rejected(self) -> None:
|
||||
"""gate_block <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_block=0)
|
||||
with pytest.raises(ValueError, match="gate_block"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_p_low_exceeds_p_high_rejected(self) -> None:
|
||||
"""gate_p_low >= gate_p_high 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_p_low=0.9, gate_p_high=0.1)
|
||||
with pytest.raises(ValueError, match="gate_p_low"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_probe_quota_negative_rejected(self) -> None:
|
||||
"""gate_probe_quota < 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_probe_quota=-0.1)
|
||||
with pytest.raises(ValueError, match="gate_probe_quota"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_gamma_decay_zero_rejected(self) -> None:
|
||||
"""gate_gamma_decay <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_gamma_decay=0.0)
|
||||
with pytest.raises(ValueError, match="gate_gamma_decay"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_gamma_decay_one_rejected(self) -> None:
|
||||
"""gate_gamma_decay >= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_gamma_decay=1.0)
|
||||
with pytest.raises(ValueError, match="gate_gamma_decay"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_cooldown_steps_zero_rejected(self) -> None:
|
||||
"""gate_cooldown_steps < 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_cooldown_steps=0)
|
||||
with pytest.raises(ValueError, match="gate_cooldown_steps"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_guard_err_zero_rejected(self) -> None:
|
||||
"""gate_guard_err <= 0 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_guard_err=0.0)
|
||||
with pytest.raises(ValueError, match="gate_guard_err"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_guard_err_one_rejected(self) -> None:
|
||||
"""gate_guard_err >= 1 应抛出 ValueError。"""
|
||||
cfg = _make_config(gate_guard_err=1.0)
|
||||
with pytest.raises(ValueError, match="gate_guard_err"):
|
||||
_validate(cfg)
|
||||
|
||||
|
||||
# ─────────────────────── test_load_config_cli_overrides ──────────────────
|
||||
|
||||
|
||||
class TestLoadConfigCliOverrides:
|
||||
"""load_config 的 CLI 覆盖层测试。"""
|
||||
|
||||
def test_cli_overrides_yaml_values(self, tmp_path: Path) -> None:
|
||||
"""CLI 参数应覆盖 YAML 中的同名字段。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
cfg = load_config(yaml_path, cli_overrides={"concurrency": 4, "max_steps": 30})
|
||||
assert cfg.concurrency == 4
|
||||
assert cfg.max_steps == 30
|
||||
|
||||
def test_cli_none_values_ignored(self, tmp_path: Path) -> None:
|
||||
"""CLI 中值为 None 的字段不应覆盖 YAML。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
cfg = load_config(yaml_path, cli_overrides={"concurrency": None, "max_steps": 20})
|
||||
assert cfg.concurrency == 12 # YAML 默认值
|
||||
assert cfg.max_steps == 20
|
||||
|
||||
def test_cli_run_id_override(self, tmp_path: Path) -> None:
|
||||
"""CLI 可通过 run_id 覆盖默认空字符串。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
cfg = load_config(
|
||||
yaml_path,
|
||||
cli_overrides={"mode": "diagnose", "run_id": "run-abc-123"},
|
||||
)
|
||||
assert cfg.run_id == "run-abc-123"
|
||||
assert cfg.mode == "diagnose"
|
||||
|
||||
def test_path_fields_converted_to_path(self, tmp_path: Path) -> None:
|
||||
"""workspace_dir 和 store_dir 应被转换为 Path 对象。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
cfg = load_config(yaml_path)
|
||||
assert isinstance(cfg.workspace_dir, Path)
|
||||
assert isinstance(cfg.store_dir, Path)
|
||||
|
||||
def test_unknown_cli_keys_ignored(self, tmp_path: Path) -> None:
|
||||
"""YAML 和 RunConfig 中不存在的 CLI key 应被忽略。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
cfg = load_config(yaml_path, cli_overrides={"nonexistent_field": 42})
|
||||
assert cfg.concurrency == 12 # 正常字段不受影响
|
||||
|
||||
def test_validation_runs_after_loading(self, tmp_path: Path) -> None:
|
||||
"""load_config 加载后应运行校验，非法值应报错。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
harness_data["mode"] = "invalid_mode"
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
with pytest.raises(ValueError, match="mode"):
|
||||
load_config(yaml_path)
|
||||
|
||||
|
||||
# ──────────────────── test_load_config_env_overrides ─────────────────────
|
||||
|
||||
|
||||
class TestLoadConfigEnvOverrides:
|
||||
"""load_config 的 .env 环境变量覆盖层测试。"""
|
||||
|
||||
def test_env_overrides_yaml_workspace_dir(
|
||||
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""环境变量 HARNESS_WORKSPACE_DIR 应覆盖 YAML 中的 workspace_dir。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
monkeypatch.setenv("HARNESS_WORKSPACE_DIR", "/custom/workspace")
|
||||
cfg = load_config(yaml_path)
|
||||
assert cfg.workspace_dir == Path("/custom/workspace")
|
||||
|
||||
def test_env_overrides_yaml_store_dir(
|
||||
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""环境变量 HARNESS_STORE_DIR 应覆盖 YAML 中的 store_dir。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
monkeypatch.setenv("HARNESS_STORE_DIR", "/custom/store")
|
||||
cfg = load_config(yaml_path)
|
||||
assert cfg.store_dir == Path("/custom/store")
|
||||
|
||||
def test_cli_overrides_env(self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
"""CLI 优先级高于 .env：CLI > .env > YAML。"""
|
||||
harness_data = _yaml_harness_dict()
|
||||
yaml_path = _write_yaml(tmp_path, harness_data)
|
||||
|
||||
monkeypatch.setenv("HARNESS_WORKSPACE_DIR", "/env/workspace")
|
||||
cfg = load_config(yaml_path, cli_overrides={"workspace_dir": "/cli/workspace"})
|
||||
assert cfg.workspace_dir == Path("/cli/workspace")
|
||||
|
||||
|
||||
# ─────────────────────────── test_val_size_floor ─────────────────────────
|
||||
|
||||
|
||||
class TestValSizeFloor:
|
||||
"""val_size >= eval_min_per_class * 11 的下限校验。"""
|
||||
|
||||
def test_val_size_below_floor_rejected(self) -> None:
|
||||
"""val_size < eval_min_per_class * 11 应抛出 ValueError。
|
||||
|
||||
eval_min_per_class=3 → 下限 = 3 * 11 = 33，val_size=30 不足。
|
||||
"""
|
||||
cfg = _make_config(eval_min_per_class=3, val_size=30)
|
||||
with pytest.raises(ValueError, match="val_size"):
|
||||
_validate(cfg)
|
||||
|
||||
def test_val_size_at_floor_accepted(self) -> None:
|
||||
"""val_size == eval_min_per_class * 11 应通过。"""
|
||||
cfg = _make_config(eval_min_per_class=3, val_size=33)
|
||||
_validate(cfg)
|
||||
|
||||
def test_val_size_above_floor_accepted(self) -> None:
|
||||
"""val_size > eval_min_per_class * 11 应通过。"""
|
||||
cfg = _make_config(eval_min_per_class=2, val_size=100)
|
||||
_validate(cfg)
|
||||
|
||||
def test_default_yaml_values_satisfy_floor(self) -> None:
|
||||
"""default.yaml 的默认值（val_size=30, eval_min_per_class=2）应满足下限。
|
||||
|
||||
下限 = 2 * 11 = 22，val_size=30 >= 22，通过。
|
||||
"""
|
||||
cfg = _make_config()
|
||||
_validate(cfg) # 不应抛出异常
|
||||
Reference in New Issue
Block a user