guzhujushi
5319589dfd
审查并精简 ChatGPT 的建库建议后落地的第一版骨架，只定接口不写实现。 - 建议审查：14 个包 → 7 个；约 250 个函数 → 约 40 个；5 个 Result 类 → 1 个 （docs/01-suggestion-review.md） - docs/：架构、路线图（含永久不做清单）、环境、Git 工作流、函数卡片模板 - src/mcm_toolkit/：core / data / stats / models / solve / decision / viz 共 23 个模块，均为签名 + docstring + NotImplementedError - tests/：9 个文件，其中 test_api_surface.py 已通过（20 passed） - examples/ problems/ notebooks/ figures/ tables/ 骨架 - pyproject.toml（核心依赖 4 个 + 6 组 extras）、environment.yml - scripts/：setup-venv.ps1（ASCII + LF）、setup-conda.ps1、doctor.ps1 - 环境：.venv（system-site-packages）与 conda 环境 mcm-toolkit 均已建好， 数值栈版本与 base 完全对齐（numpy 2.4.6 / scipy 1.18.1 / pandas 3.0.6，MKL 后端）
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""core 包测试。
|
|
|
|
TODO(L0)：实现后去掉 skip 标记。
|
|
"""
|
|
|
|
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]])
|
|
|
|
|
|
@pytest.mark.skip(reason="未实现：见 docs/03-roadmap.md L0")
|
|
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
|