mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-05 09:17:47 +02:00
Eugene YurtsevandGitHub
3e1bbd3123
* add handling for blocks that use magic commands like %pip to convert them into bash * Apply new logic to another notebook
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
import nbformat
|
|
|
|
from _scripts.notebook_convert import md_executable
|
|
|
|
|
|
EXPECTED_OUTPUT = """\
|
|
```python exec="on" source="above" session="1" result="ansi"
|
|
print("Hello, world!")
|
|
```
|
|
"""
|
|
|
|
|
|
def test_convert_normal_code_block() -> None:
|
|
notebook = nbformat.v4.new_notebook()
|
|
notebook.metadata.language_info = {"name": "python", "version": "3.11"}
|
|
notebook.cells.append(nbformat.v4.new_code_cell('print("Hello, world!")'))
|
|
markdown, _ = md_executable.from_notebook_node(notebook)
|
|
assert markdown == EXPECTED_OUTPUT
|
|
|
|
|
|
# We treat cell magic as a non-executable code block.
|
|
CELL_MAGIC_INPUT = """\
|
|
%%capture
|
|
%pip install numpy
|
|
"""
|
|
|
|
CELL_MAGIC_OUTPUT = """\
|
|
```shell
|
|
pip install numpy
|
|
```
|
|
"""
|
|
|
|
|
|
def test_convert_cell_magic() -> None:
|
|
notebook = nbformat.v4.new_notebook()
|
|
notebook.metadata.language_info = {"name": "python", "version": "3.11"}
|
|
notebook.cells.append(nbformat.v4.new_code_cell(CELL_MAGIC_INPUT))
|
|
markdown, _ = md_executable.from_notebook_node(notebook)
|
|
assert markdown == CELL_MAGIC_OUTPUT
|
|
|
|
|
|
STDIN_INPUT = """\
|
|
input("Enter your name: ")\
|
|
"""
|
|
|
|
STDIN_OUTPUT = """\
|
|
```python
|
|
input("Enter your name: ")
|
|
```
|
|
"""
|
|
|
|
|
|
def test_convert_input_cell() -> None:
|
|
notebook = nbformat.v4.new_notebook()
|
|
notebook.metadata.language_info = {"name": "python", "version": "3.11"}
|
|
notebook.cells.append(nbformat.v4.new_code_cell(STDIN_INPUT))
|
|
markdown, _ = md_executable.from_notebook_node(notebook)
|
|
assert markdown == STDIN_OUTPUT
|