first commit

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

@@ -0,0 +1,57 @@
+def cell_is_deletable(nb, index):
+    JS = f'return Jupyter.notebook.get_cell({index}).is_deletable();'
+    return nb.browser.execute_script(JS)
+
+def remove_all_cells(notebook):
+    for i in range(len(notebook.cells)):
+        notebook.delete_cell(0)
+
+INITIAL_CELLS = ['print("a")', 'print("b")', 'print("c")']
+
+def test_delete_cells(prefill_notebook):
+    a, b, c = INITIAL_CELLS
+    notebook = prefill_notebook(INITIAL_CELLS)
+
+    # Validate initial state
+    assert notebook.get_cells_contents() == [a, b, c]
+    for cell in range(0, 3):
+        assert cell_is_deletable(notebook, cell)
+
+    notebook.set_cell_metadata(0, 'deletable', 'false')
+    notebook.set_cell_metadata(1, 'deletable', 0
+    )
+    assert not cell_is_deletable(notebook, 0)
+    assert cell_is_deletable(notebook, 1)
+    assert cell_is_deletable(notebook, 2)
+
+    # Try to delete cell a (should not be deleted)
+    notebook.delete_cell(0)
+    assert notebook.get_cells_contents() == [a, b, c]
+
+    # Try to delete cell b (should succeed)
+    notebook.delete_cell(1)
+    assert notebook.get_cells_contents() == [a, c]
+
+    # Try to delete cell c (should succeed)
+    notebook.delete_cell(1)
+    assert notebook.get_cells_contents() == [a]
+
+    # Change the deletable state of cell a
+    notebook.set_cell_metadata(0, 'deletable', 'true')
+
+    # Try to delete cell a (should succeed)
+    notebook.delete_cell(0)
+    assert len(notebook.cells) == 1 # it contains an empty cell
+
+    # Make sure copied cells are deletable
+    notebook.edit_cell(index=0, content=a)
+    notebook.set_cell_metadata(0, 'deletable', 'false')
+    assert not cell_is_deletable(notebook, 0)
+    notebook.to_command_mode()
+    notebook.current_cell.send_keys('cv')
+    assert len(notebook.cells) == 2
+    assert cell_is_deletable(notebook, 1)
+
+    notebook.set_cell_metadata(0, 'deletable', 'true')  # to perform below test, remove all the cells
+    remove_all_cells(notebook)
+    assert len(notebook.cells) == 1    # notebook should create one automatically on empty notebook