fix simpel dashboard test

oegedijk · Jul 8, 2021 · dd8bb46 · dd8bb46
1 parent e1ef1f1
commit dd8bb46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration_tests/test_dashboards.py b/tests/integration_tests/test_dashboards.py
@@ -147,7 +147,7 @@ def test_simple_classification_dashboard(dash_duo):
     assert html.startswith('\n<!DOCTYPE html>\n<html'), "failed to generate dashboard to_html"
 
     dash_duo.start_server(db.app)
-    dash_duo.wait_for_text_to_equal("#simple-classifier-composite-title", "testing", timeout=20)
+    dash_duo.wait_for_text_to_equal("h1", "testing", timeout=20)
     assert dash_duo.get_logs() == [], "browser console should contain no error"
 
 
@@ -158,7 +158,7 @@ def test_simple_regression_dashboard(dash_duo):
     assert html.startswith('\n<!DOCTYPE html>\n<html'), "failed to generate dashboard to_html"
 
     dash_duo.start_server(db.app)
-    dash_duo.wait_for_text_to_equal("#simple-regression-composite-title", "testing", timeout=20)
+    dash_duo.wait_for_text_to_equal("h1", "testing", timeout=20)
     assert dash_duo.get_logs() == [], "browser console should contain no error"