diff --git a/src/tests/test_ai_tui_agent_harness.py b/src/tests/test_ai_tui_agent_harness.py index ed228a38..c01b4c26 100644 --- a/src/tests/test_ai_tui_agent_harness.py +++ b/src/tests/test_ai_tui_agent_harness.py @@ -6,6 +6,8 @@ import sys import pytest +pytest.importorskip("pty") + def _load_harness_module(): repo_root = Path(__file__).resolve().parents[2] diff --git a/test_replace8.py b/test_replace8.py new file mode 100644 index 00000000..ba0a5342 --- /dev/null +++ b/test_replace8.py @@ -0,0 +1,13 @@ +import re + +with open("src/tests/test_ai_tui_agent_harness.py", "r") as f: + content = f.read() + +new_content = re.sub( + r"(import sys\n\nimport pytest)", + r"import sys\n\nimport pytest\n\npytest.importorskip('pty')", + content +) + +with open("src/tests/test_ai_tui_agent_harness.py", "w") as f: + f.write(new_content)