Function _ctx

tests/_code_mode/test_cells_view.py:40–49

Build an AsyncCodeModeContext with a document snapshot from the kernel.

(k: Kernel)

@contextmanager
def _ctx(k: Kernel) -> Generator[AsyncCodeModeContext, None, None]:
    """Build an AsyncCodeModeContext with a document snapshot from the kernel."""
    doc = NotebookDocument(
        [
            NotebookCell(id=cid, code=cell.code, name="", config=cell.config)
            for cid, cell in k.graph.cells.items()
        ]
    )
    with notebook_document_context(doc):
        yield AsyncCodeModeContext(k)


class TestCellsViewIndex: