Uploaded Test files

venv/Lib/site-packages/notebook/tests/selenium/test_merge_cells.py

@@ -0,0 +1,36 @@
+"""Tests the merge cell api."""
+
+INITIAL_CELLS = [
+    "foo = 5",
+    "bar = 10",
+    "baz = 15",
+    "print(foo)",
+    "print(bar)",
+    "print(baz)",
+]
+
+def test_merge_cells(prefill_notebook):
+    notebook = prefill_notebook(INITIAL_CELLS)
+    a, b, c, d, e, f = INITIAL_CELLS
+
+    # Before merging, there are 6 separate cells
+    assert notebook.get_cells_contents() == [a, b, c, d, e, f]
+
+    # Focus on the second cell and merge it with the cell above
+    notebook.focus_cell(1)
+    notebook.browser.execute_script("Jupyter.notebook.merge_cell_above();")
+    merged_a_b = "%s\n\n%s" % (a, b)
+    assert notebook.get_cells_contents() == [merged_a_b, c, d, e, f]
+
+    # Focus on the second cell and merge it with the cell below
+    notebook.focus_cell(1)
+    notebook.browser.execute_script("Jupyter.notebook.merge_cell_below();")
+    merged_c_d = "%s\n\n%s" % (c, d)
+    assert notebook.get_cells_contents() == [merged_a_b, merged_c_d, e, f]
+
+    # Merge everything down to a single cell with selected cells
+    notebook.select_cell_range(0,3)
+    notebook.browser.execute_script("Jupyter.notebook.merge_selected_cells();")
+    merged_all = "%s\n\n%s\n\n%s\n\n%s" % (merged_a_b, merged_c_d, e, f)
+    assert notebook.get_cells_contents() == [merged_all]
+