diff --git a/tests/test_coder.py b/tests/test_coder.py index c6bb655a5..b3fcdeaa4 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -20,8 +20,9 @@ class TestCoder(unittest.TestCase): # Call the check_for_file_mentions method result = coder.check_for_file_mentions("Please check file1.txt and file2.py") - # Check if the result is as expected - self.assertEqual(result, "Added files: file1.txt, 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()