From 7e75cfb1fae951ec60888d7b315a1e2c613b5c16 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Sat, 27 May 2023 08:29:05 -0700
Subject: [PATCH] added missing import

---
 tests/test_coder.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/test_coder.py b/tests/test_coder.py
index b3fcdeaa4..0995b0fda 100644
--- a/tests/test_coder.py
+++ b/tests/test_coder.py
@@ -1,7 +1,9 @@
+import os
 import unittest
 from unittest.mock import MagicMock
 from aider.coder import Coder
 
+
 class TestCoder(unittest.TestCase):
     def test_check_for_file_mentions(self):
         # Mock the IO object
@@ -18,11 +20,12 @@ class TestCoder(unittest.TestCase):
         coder.repo = mock_repo
 
         # Call the check_for_file_mentions method
-        result = coder.check_for_file_mentions("Please check file1.txt and file2.py")
+        coder.check_for_file_mentions("Please check file1.txt and file2.py")
 
         # Check if coder.abs_fnames contains both files
         expected_files = {os.path.abspath("file1.txt"), os.path.abspath("file2.py")}
         self.assertEqual(coder.abs_fnames, expected_files)
 
+
 if __name__ == "__main__":
     unittest.main()