Function _ctx

tests/_code_mode/test_code_mode_context.py:33–55

Build an AsyncCodeModeContext with a document snapshot from the kernel. `extra_doc_cells` adds cells to the document that are *not* in the kernel graph, simulating cells that exist on disk but were never run.

(
    k: Kernel,
    extra_doc_cells: list[NotebookCell] | None = None,
)

@contextmanager
def _ctx(
    k: Kernel,
    extra_doc_cells: list[NotebookCell] | None = None,
) -> Generator[AsyncCodeModeContext, None, None]:
    """Build an AsyncCodeModeContext with a document snapshot from the kernel.

    `extra_doc_cells` adds cells to the document that are *not* in the
    kernel graph, simulating cells that exist on disk but were never run.
    """
    cells = [
        NotebookCell(id=cid, code=cell.code, name="", config=cell.config)
        for cid, cell in k.graph.cells.items()
    ]
    if extra_doc_cells:
        cells.extend(extra_doc_cells)
    doc = NotebookDocument(cells)
    with notebook_document_context(doc):
        # Staleness check coverage lives in test_staleness.py.
        ctx = AsyncCodeModeContext(k, skip_staleness_check=True)
        # Use a deterministic seed in tests for snapshot stability.
        ctx._id_generator = CellIdGenerator(seed=7)
        ctx._id_generator.seen_ids = set(doc.cell_ids)
        yield ctx


def _tx_ops(k: Kernel) -> list[dict[str, object]]: