fix(pytest): put user output in clean output widget

widgetti · Jun 16, 2023 · bf8d147 · bf8d147
1 parent 6f0b4bc
commit bf8d147
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 2 deletions.
diff --git a/solara/test/pytest_plugin.py b/solara/test/pytest_plugin.py
@@ -175,18 +175,21 @@ def solara_test(solara_server, solara_app, page_session: "playwright.sync_api.Pa
         assert len(keys) == 1, "expected only one context, got %s" % keys
         context = solara.server.app.contexts[keys[0]]
         with context:
+            test_output_warmup = widgets.Output()
             test_output = widgets.Output()
             page_session.locator("text=Test in solara").wait_for()
-            context.container.children[0].children[1].children[1].children = [test_output]  # type: ignore
+            context.container.children[0].children[1].children[1].children = [test_output_warmup]  # type: ignore
             try:
-                with test_output:
+                with test_output_warmup:
                     warmup()
                     button = page_session.locator(".solara-warmup-widget")
                     button.wait_for()
                     page_session.evaluate("document.fonts.ready")
                     button.click()
                     button.wait_for(state="detached")
                     page_session.evaluate("document.fonts.ready")
+                context.container.children[0].children[1].children[1].children = [test_output]  # type: ignore
+                with test_output:
                     yield
             finally:
                 test_output.close()

diff --git a/...tex_test.py/test_widget_latex_solara-flask-chromium-linux-changed-reference.png b/...tex_test.py/test_widget_latex_solara-flask-chromium-linux-changed-reference.png
diff --git a/...est.py/test_widget_latex_solara-flask-chromium-linux-ipywidgets-7-reference.png b/...est.py/test_widget_latex_solara-flask-chromium-linux-ipywidgets-7-reference.png
diff --git a/...solara-flask-chromium-linux-reference.png → ...chromium-linux-ipywidgets-8-reference.png b/...solara-flask-chromium-linux-reference.png → ...chromium-linux-ipywidgets-8-reference.png
diff --git a/...test.py/test_widget_latex_solara-starlette-chromium-linux-changed-reference.png b/...test.py/test_widget_latex_solara-starlette-chromium-linux-changed-reference.png
diff --git a/...n/latex_test.py/test_widget_latex_solara-starlette-chromium-linux-reference.png b/...n/latex_test.py/test_widget_latex_solara-starlette-chromium-linux-reference.png