mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 20:35:00 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from pathlib import Path
|
|
|
|
from aider.io import InputOutput
|
|
from aider.watch import FileWatcher
|
|
|
|
|
|
def test_ai_comment_pattern():
|
|
# Create minimal IO and Coder instances for testing
|
|
class MinimalCoder:
|
|
def __init__(self, io):
|
|
self.io = io
|
|
self.root = "."
|
|
self.abs_fnames = set()
|
|
|
|
def get_rel_fname(self, fname):
|
|
return fname
|
|
|
|
io = InputOutput(pretty=False, fancy_input=False, yes=False)
|
|
coder = MinimalCoder(io)
|
|
watcher = FileWatcher(coder)
|
|
fixtures_dir = Path(__file__).parent.parent / "fixtures"
|
|
|
|
# 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"
|
|
|
|
# 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"
|