Update assert statement to use a list for more descriptive errors and…

… less requests to the browser
jupyter · Apr 28, 2018 · b25a142 · b25a142
1 parent 17e46a7
commit b25a142
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/notebook/tests/selenium/test_dualmode_insertcell.py b/notebook/tests/selenium/test_dualmode_insertcell.py
@@ -54,7 +54,4 @@ def test_insert_cell(notebook):
     notebook.focus_cell(2)
     shift(notebook.browser, Keys.DOWN)
     notebook.current_cell.send_keys('b')
-    assert notebook.get_cell_contents(1) == ""
-    assert notebook.get_cell_contents(2) == "cell1"
-    assert notebook.get_cell_contents(3) == "cell2"
-    assert notebook.get_cell_contents(4) == ""
+    assert notebook.get_cells_contents()[1:5] == ["", "cell1", "cell2", ""]