diff --git a/tests/test_main.py b/tests/test_main.py index 4c06c7968..5c876684a 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,7 +1,6 @@ import os import sys import tempfile -from contextlib import redirect_stdin from unittest import TestCase from aider.main import main @@ -10,5 +9,5 @@ class TestMain(TestCase): with tempfile.TemporaryDirectory() as temp_dir: os.chdir(temp_dir) with open(os.devnull, 'r') as dev_null: - with redirect_stdin(dev_null): - main() + sys.stdin.close() + main()