@marimo-team/islands 0.25.3-dev191 → 0.25.3-dev193

package/src/core/cells/__tests__/cells.test.ts

@@ -36,11 +36,6 @@ import {
   type NotebookState,
   notebookAtom,
 } from "../cells";
-import {
-  focusAndScrollCellIntoView,
-  scrollToBottom,
-  scrollToTop,
-} from "../scrollCellIntoView";
 import {
   type CellData,
   createCell,
@@ -67,7 +62,6 @@ vi.mock("../scrollCellIntoView", async (importOriginal) => {
 });
 
 const FIRST_COLUMN = 0;
-const SECOND_COLUMN = 1;
 
 const { initialNotebookState, reducer, createActions } = exportedForTesting;
 
@@ -203,10 +197,6 @@ describe("cell reducer", () => {
       cellRuntime: { ...state.cellRuntime, [id]: createCellRuntimeState() },
       cellHandles: { ...state.cellHandles, [id]: createRef() },
       scrollKey: autoFocus ? id : null,
-      untouchedNewCells:
-        hideCode && autoFocus
-          ? new Set([...state.untouchedNewCells, id])
-          : state.untouchedNewCells,
     };
     mountEditors();
     return id;
@@ -247,306 +237,6 @@ describe("cell reducer", () => {
     `);
   });
 
-  it("can move a cell", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-      "
-    `);
-
-    // move first cell to the end
-    actions.moveCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [0] ''
-      "
-    `);
-
-    // move it back
-    actions.moveCell({
-      cellId: firstCellId,
-      before: true,
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-      "
-    `);
-
-    // Add a column breakpoint to test left/right movement
-    actions.addColumnBreakpoint({ cellId: cellId("1") });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-      [0] ''
-
-      > col 1
-      [1] ''
-      "
-    `);
-
-    // Move cell right
-    actions.moveCell({
-      cellId: firstCellId,
-      direction: "right",
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-
-
-      > col 1
-      [0] ''
-
-      [1] ''
-      "
-    `);
-
-    // Move cell left
-    actions.moveCell({
-      cellId: firstCellId,
-      direction: "left",
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-      [0] ''
-
-      > col 1
-      [1] ''
-      "
-    `);
-
-    // Try to move cell left when it's already in leftmost column (should noop)
-    actions.moveCell({
-      cellId: firstCellId,
-      direction: "left",
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-      [0] ''
-
-      > col 1
-      [1] ''
-      "
-    `);
-
-    // Try to move cell right when it's already in rightmost column (should noop)
-    actions.moveCell({
-      cellId: cellId("1"),
-      direction: "right",
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-      [0] ''
-
-      > col 1
-      [1] ''
-      "
-    `);
-  });
-
-  it("can drag and drop a cell", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-
-      [2] ''
-      "
-    `);
-
-    // drag first cell to the end
-    actions.dropCellOverCell({
-      cellId: firstCellId,
-      overCellId: cellId("2"),
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [2] ''
-
-      [0] ''
-      "
-    `);
-
-    // drag it back to the middle
-    actions.dropCellOverCell({
-      cellId: firstCellId,
-      overCellId: cellId("2"),
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [0] ''
-
-      [2] ''
-      "
-    `);
-  });
-
-  it("can move a cell to an exact index within a column", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-    });
-
-    const columnId = state.cellIds.atOrThrow(0).id;
-    actions.moveCellToIndex({
-      cellId: firstCellId,
-      columnId,
-      index: 3,
-    });
-
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [2] ''
-
-      [0] ''
-      "
-    `);
-  });
-
-  it("can move a cell to an exact index across columns", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-    });
-    actions.addColumnBreakpoint({ cellId: cellId("1") });
-
-    const secondColumnId = state.cellIds.atOrThrow(1).id;
-    actions.moveCellToIndex({
-      cellId: firstCellId,
-      columnId: secondColumnId,
-      index: 1,
-    });
-
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-
-
-      > col 1
-      [1] ''
-
-      [0] ''
-
-      [2] ''
-      "
-    `);
-  });
-
-  it("moveCellToIndex is a no-op when moving to the same position", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-    });
-
-    const columnId = state.cellIds.atOrThrow(0).id;
-    const before = formatCells(state);
-
-    actions.moveCellToIndex({
-      cellId: firstCellId,
-      columnId,
-      index: 0,
-    });
-
-    expect(formatCells(state)).toBe(before);
-  });
-
-  it("moveCellToIndex is a no-op for an invalid columnId", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-
-    const before = formatCells(state);
-
-    actions.moveCellToIndex({
-      cellId: firstCellId,
-      columnId: "nonexistent-column" as CellColumnId,
-      index: 0,
-    });
-
-    expect(formatCells(state)).toBe(before);
-  });
-
-  it("can move multiple cells relative to target", () => {
-    seedCell({
-      cellId: firstCellId,
-      before: false,
-    });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-
-      [2] ''
-      "
-    `);
-
-    // Move first two cells after the third
-    actions.moveCellsRelativeTo({
-      cellIds: [firstCellId, cellId("1")],
-      targetCellId: cellId("2"),
-      position: "after",
-    });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [2] ''
-
-      [0] ''
-
-      [1] ''
-      "
-    `);
-  });
-
   it("keeps error evidence after an edit and a new queued run", () => {
     const code = "raise ValueError('old failure')";
     const output: OutputMessage = {
@@ -1439,61 +1129,13 @@ describe("cell reducer", () => {
     expect(cell).toMatchSnapshot("console received while idle");
   });
 
-  it("can send a cell to the top", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-    actions.sendToTop({ cellId: cellId("2") });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [2] ''
+  it("can fold and unfold all cells", () => {
+    actions.foldAll();
+    expect(foldAllBulk).toHaveBeenCalled();
 
-      [0] ''
-
-      [1] ''
-      "
-    `);
-  });
-
-  it("can send a cell to the bottom", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-    actions.sendToBottom({ cellId: firstCellId });
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [2] ''
-
-      [0] ''
-      "
-    `);
-  });
-
-  it("can focus cells", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-
-    actions.focusCell({ cellId: cellId("1"), where: "before" });
-    expect(focusAndScrollCellIntoView).toHaveBeenCalledWith(
-      expect.objectContaining({
-        cellId: cellId("0"),
-      }),
-    );
-
-    actions.focusTopCell();
-    expect(scrollToTop).toHaveBeenCalled();
-
-    actions.focusBottomCell();
-    expect(scrollToBottom).toHaveBeenCalled();
-  });
-
-  it("can fold and unfold all cells", () => {
-    actions.foldAll();
-    expect(foldAllBulk).toHaveBeenCalled();
-
-    actions.unfoldAll();
-    expect(unfoldAllBulk).toHaveBeenCalled();
-  });
+    actions.unfoldAll();
+    expect(unfoldAllBulk).toHaveBeenCalled();
+  });
 
   it("can clear logs", () => {
     state.cellLogs = [
@@ -1514,311 +1156,6 @@ describe("cell reducer", () => {
     expect(state.cellLogs).toEqual([]);
   });
 
-  it("can collapse and expand cells", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-      code: "# Header",
-    });
-    seedCell({
-      cellId: cellId("2"),
-      before: false,
-      code: "## Subheader",
-    });
-
-    const id = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(1);
-    state.cellRuntime[id] = {
-      ...state.cellRuntime[id],
-      outline: {
-        items: [{ name: "Header", level: 1, by: { id: "header" } }],
-      },
-    };
-
-    actions.collapseCell({ cellId: id });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(id)).toBe(true);
-
-    actions.expandCell({ cellId: id });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(id)).toBe(false);
-  });
-
-  it("can collapse and expand all cells in multiple columns", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-      code: "# First Column Header",
-    });
-    seedCell({
-      cellId: cellId("2"),
-      before: false,
-      code: "## First Column Subheader",
-    });
-
-    actions.addColumnBreakpoint({ cellId: cellId("2") });
-
-    seedCell({
-      cellId: cellId("3"),
-      before: false,
-      code: "# Second Column Header",
-    });
-    seedCell({
-      cellId: cellId("4"),
-      before: false,
-      code: "## Second Column Subheader",
-    });
-
-    const firstColumnHeaderId = state.cellIds
-      .atOrThrow(FIRST_COLUMN)
-      .atOrThrow(1);
-    state.cellRuntime[firstColumnHeaderId] = {
-      ...state.cellRuntime[firstColumnHeaderId],
-      outline: {
-        items: [
-          { name: "First Column Header", level: 1, by: { id: "header" } },
-        ],
-      },
-    };
-
-    const secondColumnHeaderId = state.cellIds
-      .atOrThrow(SECOND_COLUMN)
-      .atOrThrow(0);
-    state.cellRuntime[secondColumnHeaderId] = {
-      ...state.cellRuntime[secondColumnHeaderId],
-      outline: {
-        items: [
-          { name: "Second Column Header", level: 1, by: { id: "header" } },
-        ],
-      },
-    };
-
-    actions.collapseAllCells();
-    expect(
-      state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(firstColumnHeaderId),
-    ).toBe(true);
-    expect(
-      state.cellIds.atOrThrow(SECOND_COLUMN).isCollapsed(secondColumnHeaderId),
-    ).toBe(true);
-
-    actions.expandAllCells();
-    expect(
-      state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(firstColumnHeaderId),
-    ).toBe(false);
-    expect(
-      state.cellIds.atOrThrow(SECOND_COLUMN).isCollapsed(secondColumnHeaderId),
-    ).toBe(false);
-  });
-
-  it("can collapse and expand nested cells in one call", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-      code: "# Header",
-    });
-    seedCell({
-      cellId: cellId("2"),
-      before: false,
-      code: "## Subheader",
-    });
-    seedCell({
-      cellId: cellId("3"),
-      before: false,
-      code: "### Subsubheader",
-    });
-
-    const headerId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(1);
-    state.cellRuntime[headerId] = {
-      ...state.cellRuntime[headerId],
-      outline: {
-        items: [{ name: "Header", level: 1, by: { id: "header" } }],
-      },
-    };
-
-    const subheaderId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(2);
-    state.cellRuntime[subheaderId] = {
-      ...state.cellRuntime[subheaderId],
-      outline: {
-        items: [{ name: "Subheader", level: 2, by: { id: "subheader" } }],
-      },
-    };
-
-    // Check if both the parent and child are collapsed
-    actions.collapseAllCells();
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(headerId)).toBe(
-      true,
-    );
-    actions.expandCell({ cellId: headerId });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(subheaderId)).toBe(
-      true,
-    );
-    actions.collapseCell({ cellId: headerId });
-
-    // Check if both the parent and child are expanded
-    actions.expandAllCells();
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(headerId)).toBe(
-      false,
-    );
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(subheaderId)).toBe(
-      false,
-    );
-  });
-
-  it("can collapse all cells when nested heading ranges end at the same cell", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-      code: "# Header",
-    });
-    seedCell({
-      cellId: cellId("2"),
-      before: false,
-      code: "## Subheader",
-    });
-    seedCell({
-      cellId: cellId("3"),
-      before: false,
-      code: "### Subsubheader",
-    });
-
-    const headerId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(1);
-    state.cellRuntime[headerId] = {
-      ...state.cellRuntime[headerId],
-      outline: {
-        items: [{ name: "Header", level: 1, by: { id: "header" } }],
-      },
-    };
-
-    const subheaderId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(2);
-    state.cellRuntime[subheaderId] = {
-      ...state.cellRuntime[subheaderId],
-      outline: {
-        items: [{ name: "Subheader", level: 2, by: { id: "subheader" } }],
-      },
-    };
-
-    const subsubheaderId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(3);
-    state.cellRuntime[subsubheaderId] = {
-      ...state.cellRuntime[subsubheaderId],
-      outline: {
-        items: [{ name: "Subsubheader", level: 3, by: { id: "subsubheader" } }],
-      },
-    };
-
-    // The header, subheader and subsubheader ranges all end at the last
-    // cell. Collapsing all used to throw "Node ... not found in tree" and
-    // leave the whole notebook unchanged.
-    actions.collapseAllCells();
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(headerId)).toBe(
-      true,
-    );
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).topLevelIds).toEqual([
-      firstCellId,
-      headerId,
-    ]);
-
-    // Each level is nested under its parent, collapsed
-    actions.expandCell({ cellId: headerId });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(subheaderId)).toBe(
-      true,
-    );
-    actions.expandCell({ cellId: subheaderId });
-    expect(
-      state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(subsubheaderId),
-    ).toBe(true);
-  });
-
-  it("can collapse all cells when a deep chain is followed by a later section", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({
-      cellId: cellId("1"),
-      before: false,
-      code: "# Header",
-    });
-    seedCell({
-      cellId: cellId("2"),
-      before: false,
-      code: "## Section A",
-    });
-    seedCell({
-      cellId: cellId("3"),
-      before: false,
-      code: "### Sub of A",
-    });
-    seedCell({
-      cellId: cellId("4"),
-      before: false,
-      code: "## Section B",
-    });
-
-    const headerId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(1);
-    state.cellRuntime[headerId] = {
-      ...state.cellRuntime[headerId],
-      outline: {
-        items: [{ name: "Header", level: 1, by: { id: "header" } }],
-      },
-    };
-
-    const sectionAId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(2);
-    state.cellRuntime[sectionAId] = {
-      ...state.cellRuntime[sectionAId],
-      outline: {
-        items: [{ name: "Section A", level: 2, by: { id: "section-a" } }],
-      },
-    };
-
-    const subsectionId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(3);
-    state.cellRuntime[subsectionId] = {
-      ...state.cellRuntime[subsectionId],
-      outline: {
-        items: [{ name: "Sub of A", level: 3, by: { id: "sub-of-a" } }],
-      },
-    };
-
-    const sectionBId = state.cellIds.atOrThrow(FIRST_COLUMN).atOrThrow(4);
-    state.cellRuntime[sectionBId] = {
-      ...state.cellRuntime[sectionBId],
-      outline: {
-        items: [{ name: "Section B", level: 2, by: { id: "section-b" } }],
-      },
-    };
-
-    actions.collapseAllCells();
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(headerId)).toBe(
-      true,
-    );
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).topLevelIds).toEqual([
-      firstCellId,
-      headerId,
-    ]);
-
-    // Both sections sit under the header, collapsed, with the subsection
-    // nested under section A
-    actions.expandCell({ cellId: headerId });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(sectionAId)).toBe(
-      true,
-    );
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(sectionBId)).toBe(
-      true,
-    );
-    actions.expandCell({ cellId: sectionAId });
-    expect(
-      state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(subsectionId),
-    ).toBe(true);
-  });
-
-  it("can show hidden cells", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-    actions.collapseCell({ cellId: firstCellId });
-
-    actions.showCellIfHidden({ cellId: cellId("1") });
-    expect(state.cellIds.atOrThrow(FIRST_COLUMN).isCollapsed(firstCellId)).toBe(
-      false,
-    );
-  });
-
   it("can split and undo split cells", () => {
     seedCell({
       cellId: firstCellId,
@@ -1998,79 +1335,6 @@ describe("cell reducer", () => {
     expect(state).toEqual(initialState);
   });
 
-  it("can drop a cell over another cell", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-    seedCell({ cellId: cellId("2"), before: false });
-
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-
-      [2] ''
-
-      [3] ''
-      "
-    `);
-
-    actions.dropCellOverCell({
-      cellId: cellId("0"),
-      overCellId: cellId("3"),
-    });
-
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [1] ''
-
-      [2] ''
-
-      [3] ''
-
-      [0] ''
-      "
-    `);
-  });
-
-  it("can drop a cell over a new column", () => {
-    seedCell({ cellId: firstCellId, before: false });
-    seedCell({ cellId: cellId("1"), before: false });
-
-    expect(state.cellIds.getColumns().length).toBe(1);
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      [0] ''
-
-      [1] ''
-
-      [2] ''
-      "
-    `);
-
-    actions.dropOverNewColumn({ cellId: cellId("1") });
-
-    expect(state.cellIds.getColumns().length).toBe(2);
-    expect(formatCells(state)).toMatchInlineSnapshot(`
-      "
-      > col 0
-      [0] ''
-
-      [2] ''
-
-      > col 1
-      [1] ''
-      "
-    `);
-
-    // Check that the cells are in the correct columns
-    expect(state.cellIds.getColumns()[0].topLevelIds).toEqual([
-      cellId("0"),
-      cellId("2"),
-    ]);
-    expect(state.cellIds.getColumns()[1].topLevelIds).toEqual([cellId("1")]);
-  });
-
   it("can drop a column over another column", () => {
     seedCell({ cellId: firstCellId, before: false });
     seedCell({ cellId: cellId("1"), before: false });
@@ -2129,7 +1393,11 @@ describe("cell reducer", () => {
     seedCell({ cellId: cellId("1"), before: false });
     actions.addColumnBreakpoint({ cellId: cellId("1") });
     actions.addColumnBreakpoint({ cellId: cellId("2") });
-    actions.dropOverNewColumn({ cellId: cellId("2") });
+    // A column emptied by a drag stays until the drop compacts it.
+    state = {
+      ...state,
+      cellIds: state.cellIds.addColumn(state.cellIds.getColumnIds()[1]),
+    };
 
     expect(state.cellIds.getColumns().length).toBe(4);
     expect(formatCells(state)).toMatchInlineSnapshot(`
@@ -2342,279 +1610,6 @@ describe("cell reducer", () => {
     expect(state.cellRuntime[cell2Id].output).toBeNull();
   });
 
-  describe("untouched cells functionality", () => {
-    it("starts with empty untouchedNewCells set", () => {
-      expect(state.untouchedNewCells).toEqual(new Set());
-    });
-
-    it("can mark a cell as touched", () => {
-      // Create a cell with hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-
-      // Mark it as touched
-      actions.markTouched({ cellId: newCellId });
-
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-    });
-
-    it("markTouched is idempotent", () => {
-      // Create a cell with hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-
-      // Mark it as touched multiple times
-      actions.markTouched({ cellId: newCellId });
-      actions.markTouched({ cellId: newCellId });
-      actions.markTouched({ cellId: newCellId });
-
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-    });
-
-    it("can mark a non-existent cell as touched without error", () => {
-      const nonExistentCellId = cellId("non-existent");
-
-      expect(() => {
-        actions.markTouched({ cellId: nonExistentCellId });
-      }).not.toThrow();
-
-      expect(state.untouchedNewCells.has(nonExistentCellId)).toBe(false);
-    });
-
-    it("can handle multiple untouched cells", () => {
-      // Create multiple cells with hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: false,
-      });
-
-      const cellIds = state.cellIds.inOrderIds;
-      const cell1Id = cellIds[cellIds.length - 3];
-      const cell2Id = cellIds[cellIds.length - 2];
-      const cell3Id = cellIds[cellIds.length - 1];
-
-      expect(state.untouchedNewCells.has(cell1Id)).toBe(true);
-      expect(state.untouchedNewCells.has(cell2Id)).toBe(true);
-      expect(state.untouchedNewCells.has(cell3Id)).toBe(false);
-
-      // Mark one as touched
-      actions.markTouched({ cellId: cell1Id });
-
-      expect(state.untouchedNewCells.has(cell1Id)).toBe(false);
-      expect(state.untouchedNewCells.has(cell2Id)).toBe(true);
-      expect(state.untouchedNewCells.has(cell3Id)).toBe(false);
-    });
-
-    it("preserves untouched state when cells are moved", () => {
-      // Create cells with hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-
-      const cellIds = state.cellIds.inOrderIds;
-      const cell1Id = cellIds[cellIds.length - 2];
-      const cell2Id = cellIds[cellIds.length - 1];
-
-      expect(state.untouchedNewCells.has(cell1Id)).toBe(true);
-      expect(state.untouchedNewCells.has(cell2Id)).toBe(true);
-
-      // Move cell1 to the end
-      actions.moveCell({
-        cellId: cell1Id,
-        before: false,
-      });
-
-      // Both cells should still be untouched
-      expect(state.untouchedNewCells.has(cell1Id)).toBe(true);
-      expect(state.untouchedNewCells.has(cell2Id)).toBe(true);
-    });
-
-    it("untouched cell behavior with hide_code config", () => {
-      // Create a cell with hideCode and hide_code config
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-
-      // The document hides the cell's code
-      state.cellData[newCellId] = {
-        ...state.cellData[newCellId],
-        config: { ...state.cellData[newCellId].config, hide_code: true },
-      };
-
-      // Cell should be untouched and have hide_code config
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-      expect(state.cellData[newCellId].config.hide_code).toBe(true);
-
-      // Code should not be hidden because cell is untouched
-      expect(exportedForTesting.isCellCodeHidden(state, newCellId)).toBe(false);
-
-      // Mark as touched
-      actions.markTouched({ cellId: newCellId });
-
-      // Now code should be hidden
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-      expect(exportedForTesting.isCellCodeHidden(state, newCellId)).toBe(true);
-    });
-
-    it("can mark an existing cell as untouched", () => {
-      // Create a cell without hideCode (not in untouchedNewCells)
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: false,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-
-      // Mark it as untouched
-      actions.markUntouched({ cellId: newCellId });
-
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-    });
-
-    it("markUntouched is idempotent", () => {
-      // Create a cell without hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: false,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-
-      // Mark as untouched multiple times
-      actions.markUntouched({ cellId: newCellId });
-      actions.markUntouched({ cellId: newCellId });
-      actions.markUntouched({ cellId: newCellId });
-
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-    });
-
-    it("markUntouched does not affect already untouched cells", () => {
-      // Create a cell with hideCode (already in untouchedNewCells)
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: true,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-
-      // Calling markUntouched should not change anything
-      actions.markUntouched({ cellId: newCellId });
-
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-    });
-
-    it("markTouched and markUntouched can toggle cell state", () => {
-      // Create a cell without hideCode
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: false,
-      });
-
-      const newCellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-
-      // Initially not untouched
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-
-      // Mark as untouched
-      actions.markUntouched({ cellId: newCellId });
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-
-      // Mark as touched
-      actions.markTouched({ cellId: newCellId });
-      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
-
-      // Mark as untouched again
-      actions.markUntouched({ cellId: newCellId });
-      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
-    });
-
-    it("markUntouched works for markdown cell conversion scenario", () => {
-      // Simulates converting a Python cell to Markdown
-      // 1. Create a regular cell (no hideCode)
-      seedCell({
-        cellId: "__end__",
-        before: false,
-        hideCode: false,
-      });
-
-      const cellId =
-        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
-
-      // Cell starts without hide_code and not in untouchedNewCells
-      expect(state.cellData[cellId].config.hide_code).toBe(false);
-      expect(state.untouchedNewCells.has(cellId)).toBe(false);
-      expect(exportedForTesting.isCellCodeHidden(state, cellId)).toBe(false);
-
-      // 2. Convert to markdown: the document hides the code, and the cell
-      // is marked untouched
-      state.cellData[cellId] = {
-        ...state.cellData[cellId],
-        config: { ...state.cellData[cellId].config, hide_code: true },
-      };
-      actions.markUntouched({ cellId });
-
-      // Code should NOT be hidden because cell is untouched (user can edit)
-      expect(state.cellData[cellId].config.hide_code).toBe(true);
-      expect(state.untouchedNewCells.has(cellId)).toBe(true);
-      expect(exportedForTesting.isCellCodeHidden(state, cellId)).toBe(false);
-
-      // 3. User blurs the cell (markTouched)
-      actions.markTouched({ cellId });
-
-      // Now code should be hidden
-      expect(state.untouchedNewCells.has(cellId)).toBe(false);
-      expect(exportedForTesting.isCellCodeHidden(state, cellId)).toBe(true);
-    });
-  });
-
   describe("releaseCellAtoms", () => {
     it("atom families cache atoms until explicitly removed", () => {
       const { cellDataAtom, cellRuntimeAtom, cellHandleAtom } =
@@ -2639,91 +1634,6 @@ describe("cell reducer", () => {
   });
 });
 
-describe("isCellCodeHidden", () => {
-  const state = initialNotebookState();
-  const firstCellId = state.cellIds.inOrderIds[0];
-
-  it("returns false when hide_code is false and cell is not untouched", () => {
-    const testCellId = cellId("test-cell");
-    const testState: NotebookState = {
-      ...state,
-      cellData: {
-        ...state.cellData,
-        [testCellId]: {
-          ...state.cellData[firstCellId],
-          id: testCellId,
-          config: { hide_code: false, disabled: false, column: null },
-        },
-      },
-      untouchedNewCells: new Set(),
-    };
-
-    expect(exportedForTesting.isCellCodeHidden(testState, testCellId)).toBe(
-      false,
-    );
-  });
-
-  it("returns true when hide_code is true and cell is not untouched", () => {
-    const testCellId = cellId("test-cell");
-    const testState: NotebookState = {
-      ...state,
-      cellData: {
-        ...state.cellData,
-        [testCellId]: {
-          ...state.cellData[firstCellId],
-          id: testCellId,
-          config: { hide_code: true, disabled: false, column: null },
-        },
-      },
-      untouchedNewCells: new Set(),
-    };
-
-    expect(exportedForTesting.isCellCodeHidden(testState, testCellId)).toBe(
-      true,
-    );
-  });
-
-  it("returns false when hide_code is true but cell is untouched", () => {
-    const testCellId = cellId("test-cell");
-    const testState: NotebookState = {
-      ...state,
-      cellData: {
-        ...state.cellData,
-        [testCellId]: {
-          ...state.cellData[firstCellId],
-          id: testCellId,
-          config: { hide_code: true, disabled: false, column: null },
-        },
-      },
-      untouchedNewCells: new Set([testCellId]),
-    };
-
-    expect(exportedForTesting.isCellCodeHidden(testState, testCellId)).toBe(
-      false,
-    );
-  });
-
-  it("returns false when hide_code is false and cell is untouched", () => {
-    const testCellId = cellId("test-cell");
-    const testState: NotebookState = {
-      ...state,
-      cellData: {
-        ...state.cellData,
-        [testCellId]: {
-          ...state.cellData[firstCellId],
-          id: testCellId,
-          config: { hide_code: false, disabled: false, column: null },
-        },
-      },
-      untouchedNewCells: new Set([testCellId]),
-    };
-
-    expect(exportedForTesting.isCellCodeHidden(testState, testCellId)).toBe(
-      false,
-    );
-  });
-});
-
 describe("createTracebackInfoAtom", () => {
   const store = createStore();