From 8b3b4a2e26c4fca7d4c31a1928a9ce1ad2cb1a0b Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sun, 28 May 2023 17:07:05 -0700 Subject: [PATCH] Added import of unittest.mock to test_repomap.py. --- tests/test_repomap.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/test_repomap.py b/tests/test_repomap.py index 55c3b6852..c536cb641 100644 --- a/tests/test_repomap.py +++ b/tests/test_repomap.py @@ -2,7 +2,7 @@ import os import tempfile import unittest from aider.repomap import RepoMap - +from unittest.mock import patch class TestRepoMap(unittest.TestCase): def test_get_tags_map(self): @@ -55,8 +55,6 @@ def my_function(arg1, arg2): self.assertIn("my_method", result) self.assertIn("my_function", result) - from unittest.mock import patch - def test_check_for_ctags_failure(self): with patch("subprocess.run") as mock_run: mock_run.side_effect = Exception("ctags not found")