testsAndMisc/python_pkg/token_audit/tests/test_notes_link_explicit.py
Krzysztof kuhy Rudnicki 9cb4c81322
Some checks are pending
dependency freshness / dependencies on newest stable (push) Waiting to run
md-naming / markdown naming convention (push) Waiting to run
Pre-commit checks / pre-commit (${{ matrix.group }}) (mypy, mypy) (push) Waiting to run
Pre-commit checks / pre-commit (${{ matrix.group }}) (pylint, pylint) (push) Waiting to run
Pre-commit checks / pre-commit (${{ matrix.group }}) (rest, ) (push) Waiting to run
Pre-commit checks / pre-commit (${{ matrix.group }}) (shellcheck, shellcheck) (push) Waiting to run
Python tests / test (push) Waiting to run
repo-contract / repo contract (push) Waiting to run
Shell tests / shell-tests (push) Waiting to run
Shell tests / lib-tests (push) Waiting to run
Shell tests / arch-tests (push) Waiting to run
token_audit: USD per todo note; build A/B meta
Links sessions to todo notes (note_claim.sh, todo-images prompts,
fuzzy first prompt >= 0.5, handoff chains), dates completions from
the status HLC in the first Firebase dump, and adds usd_per_note,
note_linked_usd_share, notes_done(_linked) to the snapshot, a WEEKLY
section and notes.json. token_dashboard stores the build-class A/B
(agent-ab-build.jsonl) as meta ab_build.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01WzxdWWMeapxnDcgk7HSw67
2026-11-01 19:24:21 +01:00

93 lines
2.9 KiB
Python

"""Rule (a): a Bash tool_use running note_claim.sh links the session."""
from __future__ import annotations
from typing import TYPE_CHECKING
from python_pkg.token_audit import notes_backlog, notes_link, notes_scan
from python_pkg.token_audit.tests import test_notes_fixtures as fx
if TYPE_CHECKING:
from pathlib import Path
def _notes(tmp_path: Path) -> list[notes_backlog.Note]:
return notes_backlog.parse_backlog(fx.backlog(tmp_path / "B.md").read_text())
def test_claim_with_prefix_links_note(tmp_path: Path) -> None:
root = tmp_path / "projects"
fx.transcript(
root,
"s1",
[
fx.prompt("do the thing"),
fx.tool_use(
"Bash", {"command": "~/.claude/scripts/note_claim.sh bbbbbbbb"}
),
],
)
links = notes_link.link_all(notes_scan.scan_root(root), _notes(tmp_path))
assert links["s1"] == notes_link.Link((fx.NOTE_B,), "a")
def test_claim_beats_image_and_fuzzy(tmp_path: Path) -> None:
root = tmp_path / "projects"
fx.transcript(
root,
"s1",
[
fx.prompt(f"{fx.BODY_A}\n![x](todo-images/{fx.NOTE_C}/a.png)"),
fx.tool_use("Bash", {"command": f"note_claim.sh {fx.NOTE_B}"}),
],
)
links = notes_link.link_all(notes_scan.scan_root(root), _notes(tmp_path))
assert links["s1"].rule == "a"
assert links["s1"].notes == (fx.NOTE_B,)
def test_unknown_or_ambiguous_claim_is_ignored(tmp_path: Path) -> None:
root = tmp_path / "projects"
fx.transcript(
root,
"s1",
[
fx.prompt("hello there general"),
fx.tool_use("Bash", {"command": "note_claim.sh ffffffff"}),
],
)
notes = [
*_notes(tmp_path),
notes_backlog.Note("aaaaaaaa-9999", "dup", "dup", "low", "todo", None),
]
facts = notes_scan.scan_root(root)
assert "s1" not in notes_link.link_all(facts, notes)
assert notes_link.resolve("aaaaaaaa", [n.id for n in notes]) is None
def test_claim_in_tool_result_is_not_a_link(tmp_path: Path) -> None:
root = tmp_path / "projects"
ids = f"{fx.NOTE_A} {fx.NOTE_B} {fx.NOTE_C}"
fx.transcript(
root,
"s1",
[
fx.prompt("read my backlog please now"),
fx.tool_use("Read", {"file_path": "/home/kuhy/src/todo/BACKLOG.md"}),
fx.tool_result(f"note_claim.sh {fx.NOTE_A} {ids}"),
],
)
assert notes_link.link_all(notes_scan.scan_root(root), _notes(tmp_path)) == {}
def test_claim_inside_subagent_does_not_count(tmp_path: Path) -> None:
root = tmp_path / "projects"
fx.transcript(root, "s1", [fx.prompt("hello there general")])
fx.subagent(
root,
"s1",
"x",
[fx.tool_use("Bash", {"command": f"note_claim.sh {fx.NOTE_A}"})],
)
assert notes_link.link_all(notes_scan.scan_root(root), _notes(tmp_path)) == {}