mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-23 22:04:59 +00:00
45442b99e1
# Context:
USER: add another like that, but with ctags=False. just make sure the file is in the result.
ASSISTANT: I will add a new test to the `tests/test_repomap.py` file that checks if the file is in the result when `ctags=False`.
```python
tests/test_repomap.py
<<<<<<< ORIGINAL
self.assertIn("my_function", result)
if __name__ == "__main__":
unittest.main()
=======
self.assertIn("my_function", result)
def test_get_tags_map_without_ctags(self):
# Create a temporary directory with a sample Python file containing identifiers
test_file = "test_file_without_ctags.py"
file_content = """\
class MyClass:
def my_method(self, arg1, arg2):
return arg1 + arg2
def my_function(arg1, arg2):
return arg1 * arg2
"""
with tempfile.TemporaryDirectory() as temp_dir:
with open(os.path.join(temp_dir, test_file), "w") as f:
f.write(file_content)
repo_map = RepoMap(use_ctags=False, root=temp_dir)
other_files = [os.path.join(temp_dir, test_file)]
result = repo_map.get_tags_map(other_files)
# Check if the result contains the expected tags map without ctags
self.assertIn("test_file_without_ctags.py:", result)
if __name__ == "__main__":
unittest.main()
>>>>>>> UPDATED
```
This new test creates a temporary directory with a sample Python file containing a class, a method, and a function. It then checks if the `get_tags_map` function returns the expected tags map containing the file when `ctags=False`.
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
import os
|
|
import tempfile
|
|
import unittest
|
|
from aider.repomap import RepoMap
|
|
|
|
class TestRepoMap(unittest.TestCase):
|
|
def test_get_tags_map(self):
|
|
# Create a temporary directory with sample files for testing
|
|
test_files = [
|
|
"test_file1.py",
|
|
"test_file2.py",
|
|
"test_file3.md",
|
|
"test_file4.json",
|
|
]
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
for file in test_files:
|
|
with open(os.path.join(temp_dir, file), "w") as f:
|
|
f.write("")
|
|
|
|
repo_map = RepoMap(root=temp_dir)
|
|
other_files = [os.path.join(temp_dir, file) for file in test_files]
|
|
result = repo_map.get_tags_map(other_files)
|
|
|
|
# Check if the result contains the expected tags map
|
|
self.assertIn("test_file1.py:", result)
|
|
self.assertIn("test_file2.py:", result)
|
|
self.assertNotIn("test_file3.md:", result)
|
|
self.assertNotIn("test_file4.json:", result)
|
|
|
|
def test_get_tags_map_with_identifiers(self):
|
|
# Create a temporary directory with a sample Python file containing identifiers
|
|
test_file = "test_file_with_identifiers.py"
|
|
file_content = """\
|
|
class MyClass:
|
|
def my_method(self, arg1, arg2):
|
|
return arg1 + arg2
|
|
|
|
def my_function(arg1, arg2):
|
|
return arg1 * arg2
|
|
"""
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
with open(os.path.join(temp_dir, test_file), "w") as f:
|
|
f.write(file_content)
|
|
|
|
repo_map = RepoMap(root=temp_dir)
|
|
other_files = [os.path.join(temp_dir, test_file)]
|
|
result = repo_map.get_tags_map(other_files)
|
|
|
|
# Check if the result contains the expected tags map with identifiers
|
|
self.assertIn("test_file_with_identifiers.py:", result)
|
|
self.assertIn("MyClass", result)
|
|
self.assertIn("my_method", result)
|
|
self.assertIn("my_function", result)
|
|
|
|
def test_get_tags_map_without_ctags(self):
|
|
# Create a temporary directory with a sample Python file containing identifiers
|
|
test_file = "test_file_without_ctags.py"
|
|
file_content = """\
|
|
class MyClass:
|
|
def my_method(self, arg1, arg2):
|
|
return arg1 + arg2
|
|
|
|
def my_function(arg1, arg2):
|
|
return arg1 * arg2
|
|
"""
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
with open(os.path.join(temp_dir, test_file), "w") as f:
|
|
f.write(file_content)
|
|
|
|
repo_map = RepoMap(use_ctags=False, root=temp_dir)
|
|
other_files = [os.path.join(temp_dir, test_file)]
|
|
result = repo_map.get_tags_map(other_files)
|
|
|
|
# Check if the result contains the expected tags map without ctags
|
|
self.assertIn("test_file_without_ctags.py:", result)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|