dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fixed some tests

Browse source
This commit is contained in:
Paul Gauthier 2023-06-04 14:07:50 -07:00
parent 5cc28bae15
commit 676dcd8fd1
1 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/test_repomap.py
View file

@ -8,7 +8,7 @@ from aider.repomap import RepoMap
class TestRepoMap(unittest.TestCase): class TestRepoMap(unittest.TestCase):
def test_get_tags_map(self): def test_get_repo_map(self):
# Create a temporary directory with sample files for testing # Create a temporary directory with sample files for testing
test_files = [ test_files = [
"test_file1.py", "test_file1.py",
@ -24,15 +24,15 @@ class TestRepoMap(unittest.TestCase):
repo_map = RepoMap(root=temp_dir) repo_map = RepoMap(root=temp_dir)
other_files = [os.path.join(temp_dir, file) for file in test_files] other_files = [os.path.join(temp_dir, file) for file in test_files]
result = repo_map.get_tags_map(other_files) result = repo_map.get_repo_map([], other_files)
# Check if the result contains the expected tags map # Check if the result contains the expected tags map
self.assertIn("test_file1.py:", result) self.assertIn("test_file1.py", result)
self.assertIn("test_file2.py:", result) self.assertIn("test_file2.py", result)
self.assertIn("test_file3.md:", result) self.assertIn("test_file3.md", result)
self.assertIn("test_file4.json:", result) self.assertIn("test_file4.json", result)
def test_get_tags_map_with_identifiers(self): def test_get_repo_map_with_identifiers(self):
# Create a temporary directory with a sample Python file containing identifiers # Create a temporary directory with a sample Python file containing identifiers
test_file = "test_file_with_identifiers.py" test_file = "test_file_with_identifiers.py"
file_content = """\ file_content = """\
@ -50,10 +50,10 @@ def my_function(arg1, arg2):
repo_map = RepoMap(root=temp_dir) repo_map = RepoMap(root=temp_dir)
other_files = [os.path.join(temp_dir, test_file)] other_files = [os.path.join(temp_dir, test_file)]
result = repo_map.get_tags_map(other_files) result = repo_map.get_repo_map([], other_files)
# Check if the result contains the expected tags map with identifiers # Check if the result contains the expected tags map with identifiers
self.assertIn("test_file_with_identifiers.py:", result) self.assertIn("test_file_with_identifiers.py", result)
self.assertIn("MyClass", result) self.assertIn("MyClass", result)
self.assertIn("my_method", result) self.assertIn("my_method", result)
self.assertIn("my_function", result) self.assertIn("my_function", result)
@ -61,7 +61,7 @@ def my_function(arg1, arg2):
def test_check_for_ctags_failure(self): def test_check_for_ctags_failure(self):
with patch("subprocess.run") as mock_run: with patch("subprocess.run") as mock_run:
mock_run.side_effect = Exception("ctags not found") mock_run.side_effect = Exception("ctags not found")
repo_map = RepoMap(use_ctags=True) repo_map = RepoMap()
result = repo_map.check_for_ctags() result = repo_map.check_for_ctags()
self.assertFalse(result) self.assertFalse(result)
@ -75,11 +75,11 @@ def my_function(arg1, arg2):
b' status = main()$/", "kind": "variable"}' b' status = main()$/", "kind": "variable"}'
), ),
) )
repo_map = RepoMap(use_ctags=True) repo_map = RepoMap()
result = repo_map.check_for_ctags() result = repo_map.check_for_ctags()
self.assertTrue(result) self.assertTrue(result)
def test_get_tags_map_without_ctags(self): def test_get_repo_map_without_ctags(self):
# Create a temporary directory with a sample Python file containing identifiers # Create a temporary directory with a sample Python file containing identifiers
test_files = [ test_files = [
"test_file_without_ctags.py", "test_file_without_ctags.py",
@ -96,15 +96,15 @@ def my_function(arg1, arg2):
with open(os.path.join(temp_dir, file), "w") as f: with open(os.path.join(temp_dir, file), "w") as f:
f.write("") f.write("")
repo_map = RepoMap(use_ctags=False, root=temp_dir) repo_map = RepoMap(root=temp_dir)
repo_map.has_ctags = False # force it off
other_files = [os.path.join(temp_dir, file) for file in test_files] other_files = [os.path.join(temp_dir, file) for file in test_files]
result = repo_map.get_tags_map(other_files) result = repo_map.get_repo_map([], other_files)
# Check if the result contains each specific file in the expected tags map without ctags # Check if the result contains each specific file in the expected tags map without ctags
for file in test_files: for file in test_files:
self.assertIn( self.assertIn(file, result)
f"{os.path.splitext(file)[0]}.{os.path.splitext(file)[1][1:]}:", result
)
if __name__ == "__main__": if __name__ == "__main__":