mirror of
https://github.com/shareAI-lab/analysis_claude_code.git
synced 2026-09-20 12:13:38 +08:00
fix(s08): support recursive glob and conditional micro compact
This commit is contained in:
110
tests/test_s08_context_compact.py
Normal file
110
tests/test_s08_context_compact.py
Normal file
@@ -0,0 +1,110 @@
|
||||
import runpy
|
||||
import sys
|
||||
import types
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
ROOT = Path(__file__).resolve().parents[1]
|
||||
LESSON = ROOT / "s08_context_compact" / "code.py"
|
||||
|
||||
|
||||
def load_lesson(monkeypatch, workdir: Path):
|
||||
fake_anthropic = types.ModuleType("anthropic")
|
||||
fake_dotenv = types.ModuleType("dotenv")
|
||||
|
||||
class FakeAnthropic:
|
||||
def __init__(self, *args, **kwargs):
|
||||
self.messages = types.SimpleNamespace(create=None)
|
||||
|
||||
fake_anthropic.Anthropic = FakeAnthropic
|
||||
fake_dotenv.load_dotenv = lambda override=True: None
|
||||
monkeypatch.setitem(sys.modules, "anthropic", fake_anthropic)
|
||||
monkeypatch.setitem(sys.modules, "dotenv", fake_dotenv)
|
||||
monkeypatch.setenv("MODEL_ID", "test-model")
|
||||
monkeypatch.setenv("ANTHROPIC_API_KEY", "test-key")
|
||||
monkeypatch.chdir(workdir)
|
||||
return runpy.run_path(str(LESSON))
|
||||
|
||||
|
||||
def test_glob_double_star_matches_files_at_any_depth(tmp_path, monkeypatch):
|
||||
(tmp_path / "root.py").write_text("")
|
||||
(tmp_path / "one").mkdir()
|
||||
(tmp_path / "one" / "one.py").write_text("")
|
||||
(tmp_path / "one" / "two").mkdir()
|
||||
(tmp_path / "one" / "two" / "deep.py").write_text("")
|
||||
lesson = load_lesson(monkeypatch, tmp_path)
|
||||
|
||||
matches = set(lesson["run_glob"]("**/*.py").splitlines())
|
||||
|
||||
assert matches == {"root.py", "one/one.py", "one/two/deep.py"}
|
||||
|
||||
|
||||
def test_prepare_preserves_tool_results_while_context_is_within_limit(
|
||||
tmp_path, monkeypatch):
|
||||
lesson = load_lesson(monkeypatch, tmp_path)
|
||||
messages = []
|
||||
expected_results = []
|
||||
for index in range(5):
|
||||
tool_id = f"tool-{index}"
|
||||
result = f"result-{index}:" + "x" * 200
|
||||
expected_results.append(result)
|
||||
messages.extend([
|
||||
{"role": "assistant", "content": [
|
||||
{"type": "tool_use", "id": tool_id, "name": "bash", "input": {}}
|
||||
]},
|
||||
{"role": "user", "content": [
|
||||
{"type": "tool_result", "tool_use_id": tool_id, "content": result}
|
||||
]},
|
||||
])
|
||||
messages.append({"role": "assistant", "content": [
|
||||
{"type": "text", "text": "continue"}
|
||||
]})
|
||||
|
||||
prepared = lesson["COMPACTOR"].prepare(messages, "inspect the repository")
|
||||
actual_results = [
|
||||
block["content"]
|
||||
for message in prepared
|
||||
if message["role"] == "user"
|
||||
for block in message["content"]
|
||||
if block["type"] == "tool_result"
|
||||
]
|
||||
|
||||
assert actual_results == expected_results
|
||||
|
||||
|
||||
def test_prepare_micro_compacts_tool_results_after_context_exceeds_limit(
|
||||
tmp_path, monkeypatch):
|
||||
lesson = load_lesson(monkeypatch, tmp_path)
|
||||
messages = []
|
||||
for index in range(5):
|
||||
tool_id = f"tool-{index}"
|
||||
messages.extend([
|
||||
{"role": "assistant", "content": [
|
||||
{"type": "tool_use", "id": tool_id, "name": "bash", "input": {}}
|
||||
]},
|
||||
{"role": "user", "content": [
|
||||
{"type": "tool_result", "tool_use_id": tool_id,
|
||||
"content": f"result-{index}:" + "x" * 1000}
|
||||
]},
|
||||
])
|
||||
messages.append({"role": "assistant", "content": [
|
||||
{"type": "text", "text": "continue"}
|
||||
]})
|
||||
compactor = lesson["COMPACTOR"]
|
||||
compactor.CONTEXT_CHAR_LIMIT = 4500
|
||||
|
||||
prepared = compactor.prepare(messages, "inspect the repository")
|
||||
actual_results = [
|
||||
block["content"]
|
||||
for message in prepared
|
||||
if message["role"] == "user"
|
||||
for block in message["content"]
|
||||
if block["type"] == "tool_result"
|
||||
]
|
||||
|
||||
assert actual_results[:2] == [
|
||||
"[Earlier tool result omitted.]",
|
||||
"[Earlier tool result omitted.]",
|
||||
]
|
||||
assert all(result.startswith(f"result-{index}:")
|
||||
for index, result in enumerate(actual_results[2:], start=2))
|
||||
Reference in New Issue
Block a user