fix: attempt at fixing test caused by tabcomplete

ErikBjare · Oct 11, 2023 · b02804b · b02804b
1 parent 6eb763c
commit b02804b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/gptme/cli.py b/gptme/cli.py
@@ -264,12 +264,14 @@ def main(
     # init
     logger.debug("Started")
     load_dotenv()
-    register_tabcomplete()
     _load_readline_history()
     init_llm(llm)  # set up API_KEY and API_BASE
 
     if "PYTEST_CURRENT_TEST" in os.environ:
         interactive = False
+    else:
+        # for some reason it bugs out shell tests in CI
+        register_tabcomplete()
 
     if not interactive:
         no_confirm = True

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -32,7 +32,7 @@ def test_shell(name: str):
         )
         output = result.output.split("System")[-1]
         # check for two 'yes' in output (both command and stdout)
-        assert output.count("yes") == 2
+        assert output.count("yes") == 2, result.output
         assert result.exit_code == 0