style: Fix formatting and linting issues in test_main.py

This commit is contained in:
Paul Gauthier (aider) 2024-08-09 16:43:27 -04:00
parent da5ea3a767
commit 3549ed1205

View file

@ -401,18 +401,18 @@ class TestMain(TestCase):
with GitTemporaryDirectory():
test_file = "test_file.txt"
Path(test_file).touch()
coder = main(
["--read", test_file, "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
return_coder=True,
)
self.assertIn(str(Path(test_file).resolve()), coder.abs_read_only_fnames)
def test_read_option_with_external_file(self):
with tempfile.NamedTemporaryFile(mode='w', delete=False) as external_file:
with tempfile.NamedTemporaryFile(mode="w", delete=False) as external_file:
external_file.write("External file content")
external_file_path = external_file.name
@ -424,7 +424,7 @@ class TestMain(TestCase):
output=DummyOutput(),
return_coder=True,
)
self.assertIn(external_file_path, coder.abs_read_only_fnames)
finally:
os.unlink(external_file_path)