🔨 Improve wait_for_text_to_equal

plotly · Dec 28, 2018 · 9a3c0f5 · 9a3c0f5
1 parent cb74460
commit 9a3c0f5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/test_integration.py b/test/test_integration.py
@@ -57,13 +57,13 @@ def wait_for_element_by_css_selector(self, selector):
         )
 
     def wait_for_text_to_equal(self, selector, assertion_text):
+
+        def text_equal(driver):
+            return driver.find_element_by_css_selector(selector).text \
+                   == assertion_text
+
         WebDriverWait(self.driver, TIMEOUT).until(
-            EC.text_to_be_present_in_element((By.CSS_SELECTOR, selector),
-                                             assertion_text)
-        )
-        self.assertEqual(
-            assertion_text,
-            self.driver.find_element_by_css_selector(selector).text
+            text_equal
         )
 
     def snapshot(self, name):