ci: :green_heart: debugging test

1 files changed, 21 insertions, 19 deletions

diff --git a/continuedev/src/continuedev/tests/step_test.py b/continuedev/src/continuedev/tests/step_test.py
index c9bb5441..9d5f2884 100644
--- a/continuedev/src/continuedev/tests/step_test.py
+++ b/continuedev/src/continuedev/tests/step_test.py
@@ -2,37 +2,37 @@ import pytest
 
 from continuedev.core.config import ContinueConfig
 from continuedev.headless import start_headless_session
-from continuedev.plugins.steps.main import EditHighlightedCodeStep
+from continuedev.models.filesystem import Range, RangeInFileWithContents
+from continuedev.plugins.steps.chat import SimpleChatStep
 from continuedev.plugins.steps.core.core import UserInputStep
+from continuedev.plugins.steps.main import EditHighlightedCodeStep
 from continuedev.plugins.steps.on_traceback import DefaultOnTracebackStep
-from continuedev.tests.util.prompts import tokyo_test_pair, dotenv_test_pair
-from continuedev.models.filesystem import RangeInFileWithContents, Range
-from continuedev.plugins.steps.chat import SimpleChatStep
+from continuedev.tests.util.prompts import dotenv_test_pair, tokyo_test_pair
 
 TEST_CONFIG = ContinueConfig()
 
 
 @pytest.mark.asyncio
 async def test_step():
+    print("1")
     session = await start_headless_session(config=TEST_CONFIG)
 
+    print("2")
     await session.autopilot.run_from_step(UserInputStep(user_input=tokyo_test_pair[0]))
 
+    print("3")
     full_state = await session.autopilot.get_full_state()
 
-    assert (
-        isinstance(full_state.history.timeline[-1].step, SimpleChatStep)
-    )
+    assert isinstance(full_state.history.timeline[-1].step, SimpleChatStep)
 
-    assert (
-        not full_state.history.timeline[-1].step.hide
-    )
+    assert not full_state.history.timeline[-1].step.hide
 
     assert (
         full_state.history.timeline[-1].step.description.strip().lower()
         == tokyo_test_pair[1]
     )
 
+    print("4")
     await session.autopilot.cleanup()
 
 
@@ -40,12 +40,12 @@ async def test_step():
 async def test_traceback_step():
     session = await start_headless_session(config=TEST_CONFIG)
 
-    await session.autopilot.run_from_step(DefaultOnTracebackStep(output=dotenv_test_pair[0]))
+    await session.autopilot.run_from_step(
+        DefaultOnTracebackStep(output=dotenv_test_pair[0])
+    )
 
     full_state = await session.autopilot.get_full_state()
-    assert (
-        dotenv_test_pair[1] in full_state.history.timeline[-1].step.description
-    )
+    assert dotenv_test_pair[1] in full_state.history.timeline[-1].step.description
 
     await session.autopilot.cleanup()
 
@@ -54,15 +54,17 @@ async def test_traceback_step():
 async def test_edit_step():
     session = await start_headless_session(config=TEST_CONFIG)
 
-    range_in_file = RangeInFileWithContents(filepath=__file__, range=Range.from_shorthand(0, 0, 0, 0), contents="")
+    range_in_file = RangeInFileWithContents(
+        filepath=__file__, range=Range.from_shorthand(0, 0, 0, 0), contents=""
+    )
 
     await session.autopilot.handle_highlighted_code(range_in_files=[range_in_file])
 
-    await session.autopilot.run_from_step(EditHighlightedCodeStep(user_input="Don't edit this code"))
+    await session.autopilot.run_from_step(
+        EditHighlightedCodeStep(user_input="Don't edit this code")
+    )
 
     full_state = await session.autopilot.get_full_state()
-    assert (
-        isinstance(full_state.history.timeline[-1].step.description, str)
-    )
+    assert isinstance(full_state.history.timeline[-1].step.description, str)
 
     await session.autopilot.cleanup()