dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 08:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: update test to use get_ai_comments method

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-01 10:28:29 -08:00
parent e9ce04ffe9
commit 154cb463e5
1 changed files with 12 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/basic/test_watch.py
View file

@ -1,27 +1,20 @@
from pathlib import Path
from aider.watch import FileWatcher
def test_ai_comment_pattern():
# Read fixture files
# Create a FileWatcher instance for testing
watcher = FileWatcher(None)
fixtures_dir = Path(__file__).parent.parent / "fixtures"
with open(fixtures_dir / "watch.py") as f:
py_content = f.read()
with open(fixtures_dir / "watch.js") as f:
js_content = f.read()
# Test Python fixture
py_path = fixtures_dir / "watch.py"
py_lines, py_comments, py_has_bang = watcher.get_ai_comments(str(py_path))
assert len(py_lines) == 11, f"Expected 11 AI comments in Python fixture, found {len(py_lines)}"
assert py_has_bang, "Expected at least one bang (!) comment in Python fixture"
# Count AI comments in Python fixture
py_matches = 0
for line in py_content.splitlines():
if FileWatcher.ai_comment_pattern.search(line):
py_matches += 1
assert py_matches == 11, f"Expected 11 AI comments in Python fixture, found {py_matches}"
# Count AI comments in JavaScript fixture
js_matches = 0
for line in js_content.splitlines():
if FileWatcher.ai_comment_pattern.search(line):
js_matches += 1
assert js_matches == 11, f"Expected 11 AI comments in JavaScript fixture, found {js_matches}"
# Test JavaScript fixture
js_path = fixtures_dir / "watch.js"
js_lines, js_comments, js_has_bang = watcher.get_ai_comments(str(js_path))
assert len(js_lines) == 11, f"Expected 11 AI comments in JavaScript fixture, found {len(js_lines)}"
assert js_has_bang, "Expected at least one bang (!) comment in JavaScript fixture"