Zum Inhalt

tests/test_config_overrides.py

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from __future__ import annotations

import pytest

from touchstone.config import apply_overrides, load_config


def test_apply_overrides_replaces_whitelisted_keys():
    cfg = load_config("config.m5.yaml")
    out = apply_overrides(cfg, {"runs_per_cell": 4, "seed": 7})
    assert out.runs_per_cell == 4
    assert out.seed == 7
    # everything else intact
    assert out.endpoint == cfg.endpoint
    assert out.models == cfg.models
    assert out.temperature == cfg.temperature
    # original untouched (returns a copy)
    assert cfg.runs_per_cell != 4 or cfg.seed != 7


def test_apply_overrides_temperature():
    cfg = load_config("config.m5.yaml")
    out = apply_overrides(cfg, {"temperature": 0.7})
    assert out.temperature == 0.7


def test_apply_overrides_unknown_key_raises():
    cfg = load_config("config.m5.yaml")
    with pytest.raises(ValueError):
        apply_overrides(cfg, {"endpoint": {"base_url": "http://x"}})


def test_apply_overrides_empty_is_noop():
    cfg = load_config("config.m5.yaml")
    out = apply_overrides(cfg, {})
    assert out == cfg


def test_apply_overrides_rejects_bool_for_int():
    cfg = load_config("config.m5.yaml")
    with pytest.raises(ValueError):
        apply_overrides(cfg, {"runs_per_cell": True})  # bool is an int subclass — rejected


def test_apply_overrides_rejects_float_for_int():
    cfg = load_config("config.m5.yaml")
    with pytest.raises(ValueError):
        apply_overrides(cfg, {"runs_per_cell": 2.5})


def test_apply_overrides_revalidates_field_constraint():
    cfg = load_config("config.m5.yaml")
    # runs_per_cell must stay >= 2 — the re-validation through Config.model_validate enforces it.
    with pytest.raises(ValueError):
        apply_overrides(cfg, {"runs_per_cell": 1})