guzhujushi
cb96c847dd
L0 第 1 项落地。 - __repr__：单行紧凑；data 只给类型与规模（ndarray(100000)）， 绝不整数组打印；__repr__ 永不抛异常 - summary()：可直接贴进论文的 Markdown 表格， 行序 success → params.* → metrics.* → extras.*（按插入顺序，不排序）， float 走 .6g，竖线转义，message 以引用行附在表后 - 新增 5 个模块级私有辅助：_fmt_scalar / _fmt_cell / _fmt_mapping / _describe_data / _truncate - tests/test_core.py：去掉 test_result_repr_is_readable 的 skip， 并补 3 条行为测试（默认值与失败路径与十万级数组、Markdown 结构、幂等） - STATUS.md：L0 进度 1/12；新增决策 21（repr 不打印 data 内容）、 22（summary 行序不排序）；CHANGELOG.md 同步
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
"""core 包测试。
|
||
|
||
TODO(L0)：TestRandom / TestValidate 实现后去掉 skip 标记（Result 已完成）。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import numpy as np
|
||
import pytest
|
||
|
||
|
||
@pytest.mark.skip(reason="未实现：见 docs/03-roadmap.md L0")
|
||
class TestRandom:
|
||
def test_get_rng_is_reproducible(self):
|
||
from mcm_toolkit.core import get_rng
|
||
|
||
assert get_rng(42).random() == get_rng(42).random()
|
||
|
||
def test_get_rng_passes_through_generator(self):
|
||
from mcm_toolkit.core import get_rng
|
||
|
||
g = np.random.default_rng(0)
|
||
assert get_rng(g) is g
|
||
|
||
def test_set_seed_makes_global_reproducible(self):
|
||
from mcm_toolkit.core import set_seed
|
||
|
||
set_seed(7)
|
||
a = np.random.default_rng(7).random()
|
||
set_seed(7)
|
||
b = np.random.default_rng(7).random()
|
||
assert a == b
|
||
|
||
|
||
@pytest.mark.skip(reason="未实现：见 docs/03-roadmap.md L0")
|
||
class TestValidate:
|
||
def test_check_same_length_raises_with_detail(self):
|
||
from mcm_toolkit.core import check_same_length
|
||
|
||
with pytest.raises(ValueError, match=r"2.*3|长度"):
|
||
check_same_length([1, 2], [1, 2, 3])
|
||
|
||
def test_check_array_rejects_nan(self):
|
||
from mcm_toolkit.core import check_array
|
||
|
||
with pytest.raises(ValueError):
|
||
check_array([1.0, np.nan, 2.0], finite=True)
|
||
|
||
def test_check_probability_bounds(self):
|
||
from mcm_toolkit.core import check_probability
|
||
|
||
assert check_probability(0.5) == 0.5
|
||
with pytest.raises(ValueError):
|
||
check_probability(1.5)
|
||
|
||
def test_check_square_matrix(self):
|
||
from mcm_toolkit.core import check_square_matrix
|
||
|
||
check_square_matrix([[1, 2], [3, 4]])
|
||
with pytest.raises(ValueError):
|
||
check_square_matrix([[1, 2, 3], [4, 5, 6]])
|
||
|
||
|
||
def test_result_repr_is_readable():
|
||
from mcm_toolkit.core import Result
|
||
|
||
text = repr(Result("demo", params={"a": 1}, metrics={"r2": 0.9}))
|
||
# 只断言关键信息在场，不断言完整字符串——否则以后调格式会让测试集体变红
|
||
assert "demo" in text and "r2" in text
|
||
|
||
|
||
def test_result_repr_stays_short_and_safe():
|
||
"""默认值、失败结果、十万级数组：repr 都不能崩，也不能刷屏。"""
|
||
from mcm_toolkit.core import Result
|
||
|
||
assert "empty" in repr(Result("empty"))
|
||
|
||
failed = repr(Result("boom", success=False, message="求解未收敛"))
|
||
assert "success=False" in failed
|
||
assert "求解未收敛" in failed
|
||
|
||
noisy = repr(Result("mc", data=np.arange(100_000)))
|
||
assert "ndarray(100000)" in noisy
|
||
assert len(noisy) < 200, "data 不应被整个打印出来"
|
||
|
||
|
||
def test_result_summary_is_markdown_table():
|
||
from mcm_toolkit.core import Result
|
||
|
||
result = Result("topsis", params={"n": 3}, metrics={"best": 0.83})
|
||
text = result.summary()
|
||
lines = text.splitlines()
|
||
|
||
assert lines[0] == "### topsis"
|
||
assert lines[2] == "| 项 | 值 |"
|
||
assert "| success | True |" in lines
|
||
assert "| params.n | 3 |" in lines
|
||
assert "| metrics.best | 0.83 |" in lines
|
||
assert result.summary() == text, "summary 必须无副作用、可重复调用"
|
||
|
||
|
||
def test_result_summary_appends_message_on_failure():
|
||
from mcm_toolkit.core import Result
|
||
|
||
text = Result("gm11", success=False, message="样本太少").summary()
|
||
assert "| success | False |" in text
|
||
assert text.rstrip().endswith("> 样本太少")
|