add wait_for_selector to the cells property

jupyter · Mar 22, 2018 · 51c2d27 · 51c2d27
1 parent 78254f0
commit 51c2d27
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/notebook/tests/selenium/utils.py b/notebook/tests/selenium/utils.py
@@ -32,6 +32,10 @@ def body(self):
 
     @property
     def cells(self):
+        """Gets all cells once they are visible.
+        
+        """
+        wait_for_selector(self.browser, ".cell")
         return self.browser.find_elements_by_class_name("cell")