diff --git a/tests/test_rendering_text.py b/tests/test_rendering_text.py index b1a22732..d9a917f3 100644 --- a/tests/test_rendering_text.py +++ b/tests/test_rendering_text.py @@ -31,6 +31,8 @@ def go_to_app(page: Page, streamlit_app: StreamlitRunner): page.goto(streamlit_app.server_url) # Wait for app to load page.get_by_role("img", name="Running...").is_hidden() + page.get_by_text("tab1").wait_for(state="visible") + page.get_by_text("tab2").wait_for(state="visible") def test_should_render_template_check_container_size(page: Page): diff --git a/tests/test_resolution_boost.py b/tests/test_resolution_boost.py index 354f9b60..0a746a19 100644 --- a/tests/test_resolution_boost.py +++ b/tests/test_resolution_boost.py @@ -31,6 +31,8 @@ def go_to_app(page: Page, streamlit_app: StreamlitRunner): page.goto(streamlit_app.server_url) # Wait for app to load page.get_by_role("img", name="Running...").is_hidden() + page.get_by_text("tab 1").wait_for(state="visible") + page.get_by_text("tab 2").wait_for(state="visible") def test_resolution_boost(page: Page): diff --git a/tests/test_tabs.py b/tests/test_tabs.py index 3119f1e9..f7c1777c 100644 --- a/tests/test_tabs.py +++ b/tests/test_tabs.py @@ -31,6 +31,8 @@ def go_to_app(page: Page, streamlit_app: StreamlitRunner): page.goto(streamlit_app.server_url) # Wait for app to load page.get_by_role("img", name="Running...").is_hidden() + page.get_by_text("tab1").wait_for(state="visible") + page.get_by_text("tab2").wait_for(state="visible") def test_should_render_template_check_container_size(page: Page): diff --git a/tests/test_tabs_2.py b/tests/test_tabs_2.py index f9c1ce91..2d860d96 100644 --- a/tests/test_tabs_2.py +++ b/tests/test_tabs_2.py @@ -31,6 +31,8 @@ def go_to_app(page: Page, streamlit_app: StreamlitRunner): page.goto(streamlit_app.server_url) # Wait for app to load page.get_by_role("img", name="Running...").is_hidden() + page.get_by_text("tab1").wait_for(state="visible") + page.get_by_text("tab2").wait_for(state="visible") def test_should_render_template_check_container_size(page: Page):