test: tighten review test assertions

Author: clean6378-max-it

tests/test_error_codes.py

@@ -149,4 +149,4 @@ def test_cli_ambiguous_session_prefix_surfaces_code(tmp_path) -> None:
     proc = _run_cli(["stats", "--base-dir", str(base), "--session", "session_aaa"])
     assert proc.returncode == 1
     assert "SESSION_AMBIGUOUS_PREFIX" in proc.stderr
-    assert "SESSION_NOT_FOUND" not in proc.stderr.split("SESSION_AMBIGUOUS_PREFIX")[0]
+    assert "SESSION_NOT_FOUND" not in proc.stderr

tests/test_tool_dispatch_sync.py

@@ -78,9 +78,9 @@ def _parse_frontend_tool_result_renderers(path: Path) -> frozenset[str]:
 
 
 def _parse_top_level_function_body(text: str, func_name: str) -> str:
-    """Slice a module-level ``def`` through the next top-level ``def`` (if any)."""
+    """Slice a module-level ``def`` through the next top-level ``def`` or EOF."""
     match = re.search(
-        rf"^def {re.escape(func_name)}\(.*?(?=^\ndef )",
+        rf"^def {re.escape(func_name)}\(.*?(?=^\ndef |\Z)",
         text,
         re.DOTALL | re.MULTILINE,
     )