37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
"""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]
|
||
|
|
|