fix: tests in jupyter notebook being very flaky

widgetti · Sep 29, 2023 · f3f6c53 · f3f6c53
1 parent 13652f1
commit f3f6c53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/solara/test/pytest_plugin.py b/solara/test/pytest_plugin.py
@@ -424,6 +424,7 @@ def run(f: Callable):
         \n"""
         write_notebook([code_setup, code_from_function(f)], notebook_path)
         page_session.goto(base_url + f"/notebooks/notebook.ipynb?v={count}")
+        page_session.locator(".prompt_container >> nth=0").wait_for()
         page_session.locator("text=Kernel starting, please wait...").wait_for(state="detached")
         page_session.locator("Kernel Ready").wait_for(state="detached")
         page_session.locator('css=[data-jupyter-action="jupyter-notebook:run-cell-and-select-next"]').click()