Fix a false positive in integration test

mckinsey · Jul 30, 2024 · 60f7892 · 60f7892
1 parent 77fc1ac
commit 60f7892
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/vizro-ai/tests/integration/test_example.py b/vizro-ai/tests/integration/test_example.py
@@ -43,9 +43,15 @@ def test_chart_with_explanation():
     assert_that(
         resp.business_insights,
         any_of(
-            contains_string("GDP per capita in the United States"),
-            contains_string("GDP in the United States"),
-            contains_string("GDP in the US"),
+            contains_string("GDP per capita"),
+            contains_string("GDP"),
+        ),
+    )
+    assert_that(
+        resp.business_insights,
+        any_of(
+            contains_string("United States"),
+            contains_string("US"),
         ),
     )
     assert_that(