mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-20 22:52:29 +02:00
Eugene YurtsevandGitHub
c1f337f50b
* Add ast parsing to determine whether we should include result="ansi". It's not meant to be perfect, but will hopefully catch the most common cases. Still requires manual review. * Ideally we could suppress output in markdown-exec in the future.
106 lines
3.0 KiB
Python
106 lines
3.0 KiB
Python
import nbformat
|
|
import pytest
|
|
|
|
from _scripts.notebook_convert import (
|
|
_convert_links_in_markdown,
|
|
md_executable,
|
|
_has_output,
|
|
)
|
|
|
|
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
|
|
|
|
|
|
NO_STDOUT_EXPECTED = """\
|
|
```python exec="on" source="above" session="1"
|
|
display(x)
|
|
```
|
|
"""
|
|
|
|
|
|
def test_convert_block_without_output() -> None:
|
|
notebook = nbformat.v4.new_notebook()
|
|
notebook.metadata.language_info = {"name": "python", "version": "3.11"}
|
|
notebook.cells.append(nbformat.v4.new_code_cell("display(x)"))
|
|
markdown, _ = md_executable.from_notebook_node(notebook)
|
|
assert markdown == NO_STDOUT_EXPECTED
|
|
|
|
|
|
def test_has_output() -> None:
|
|
"""Test if a given code block is expected to have output."""
|
|
assert _has_output("print('Hello, world!')") is True
|
|
assert _has_output("print_stream(some_iterable)") is True
|
|
assert _has_output("foo.y") is True
|
|
assert _has_output("display(x)") is False
|
|
assert _has_output("assert 1 == 1") is False
|
|
assert _has_output("def foo(): pass") is False
|
|
assert _has_output("import foobar") is False
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"source, expected",
|
|
[
|
|
(
|
|
"This is a [link](https://example.com).",
|
|
"This is a [link](https://example.com).",
|
|
),
|
|
("This is a [link](../foo).", "This is a [link](foo.md)."),
|
|
("This is a [link](../foo#hello).", "This is a [link](foo.md#hello)."),
|
|
("This is a [link](../foo/#hello).", "This is a [link](foo.md#hello)."),
|
|
],
|
|
)
|
|
def test_link_conversion(source: str, expected: str) -> None:
|
|
"""Test logic to convert links in markdown cells."""
|
|
assert _convert_links_in_markdown(source) == expected
|