| from __future__ import annotations |
|
|
| import json |
|
|
| from wal_tat.runtime import cli |
| from wal_tat.runtime.platform import detect_platform |
|
|
|
|
| def test_doctor_is_json(capsys): |
| assert cli.main(["doctor"]) == 0 |
| payload = json.loads(capsys.readouterr().out) |
| assert payload["machine"] |
| assert payload["recommended_cpu_threads"] >= 1 |
| assert payload["recommended_device"] in {"cpu", "mps", "cuda:0"} |
|
|
|
|
| def test_inspect_minimal_checkpoint(tmp_path, capsys): |
| manifest = { |
| "variant": "test", |
| "release": {"version": "v0"}, |
| "accounting": {"unique_parameters": 10, "maximum_whole_bpw": 3.0}, |
| "base": {"matrix_count": 2}, |
| "overlays": [{"name": "x"}], |
| } |
| (tmp_path / "manifest.json").write_text(json.dumps(manifest)) |
| (tmp_path / "packed_runtime.py").write_text("# test\n") |
| assert cli.main(["inspect", str(tmp_path)]) == 0 |
| payload = json.loads(capsys.readouterr().out) |
| assert payload["packed_matrices"] == 2 |
| assert payload["overlays"] == 1 |
| assert payload["runtime_files"] == ["packed_runtime.py"] |
|
|
|
|
| def test_platform_thread_default_is_bounded(): |
| info = detect_platform() |
| assert 1 <= info.recommended_cpu_threads <= info.cpu_count |
|
|
