64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
|
"""Plan 03-02 CONN-07 RED: iai-mcp topology CLI.
|
||
|
|
|
||
|
|
The `topology` subcommand prints one key:value line per metric:
|
||
|
|
C: <float>
|
||
|
|
L: <float>
|
||
|
|
sigma: <float | "insufficient_data">
|
||
|
|
communities: <int>
|
||
|
|
rich_club_ratio: <float>
|
||
|
|
N: <int>
|
||
|
|
regime: <str>
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import re
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from iai_mcp.cli import main as cli_main
|
||
|
|
|
||
|
|
|
||
|
|
def test_topology_subcommand_registered():
|
||
|
|
"""`iai-mcp topology --help` must succeed (subparser registered)."""
|
||
|
|
with pytest.raises(SystemExit) as ex:
|
||
|
|
cli_main(["topology", "--help"])
|
||
|
|
# argparse --help calls sys.exit(0) on success
|
||
|
|
assert ex.value.code == 0
|
||
|
|
|
||
|
|
|
||
|
|
def test_topology_prints_required_keys(tmp_path, capsys, monkeypatch):
|
||
|
|
"""All seven key:value lines must appear in output."""
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
code = cli_main(["topology"])
|
||
|
|
assert code == 0
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
|
||
|
|
assert re.search(r"^C:\s", out, re.MULTILINE), f"missing 'C: ' line in {out!r}"
|
||
|
|
assert re.search(r"^L:\s", out, re.MULTILINE), f"missing 'L: ' line in {out!r}"
|
||
|
|
assert re.search(r"^sigma:\s", out, re.MULTILINE), (
|
||
|
|
f"missing 'sigma: ' line in {out!r}"
|
||
|
|
)
|
||
|
|
assert re.search(r"^communities:\s", out, re.MULTILINE), (
|
||
|
|
f"missing 'communities: ' line in {out!r}"
|
||
|
|
)
|
||
|
|
assert re.search(r"^rich_club_ratio:\s", out, re.MULTILINE), (
|
||
|
|
f"missing 'rich_club_ratio: ' line in {out!r}"
|
||
|
|
)
|
||
|
|
assert re.search(r"^N:\s", out, re.MULTILINE), f"missing 'N: ' line in {out!r}"
|
||
|
|
assert re.search(r"^regime:\s", out, re.MULTILINE), (
|
||
|
|
f"missing 'regime: ' line in {out!r}"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_topology_empty_store_prints_insufficient_data(tmp_path, capsys, monkeypatch):
|
||
|
|
"""Fresh store: N is small, sigma should print as 'insufficient_data'."""
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
code = cli_main(["topology"])
|
||
|
|
assert code == 0
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
# On an empty store, sigma must be "insufficient_data" or the regime is
|
||
|
|
# "insufficient_data" -- either way, the line must contain the marker.
|
||
|
|
assert "insufficient_data" in out, (
|
||
|
|
f"empty store must surface insufficient_data; got {out!r}"
|
||
|
|
)
|