mirror of
https://github.com/langchain-ai/kork.git
synced 2026-06-30 22:28:05 -04:00
Eugene Yurtsev
31 lines
625 B
Python
31 lines
625 B
Python
from kork.ast_printer import AstPrinter
|
|
from kork.examples import c_, format_examples, r_
|
|
|
|
|
|
def add_(x: int, y: int) -> int:
|
|
"""Add two numbers."""
|
|
return x + y
|
|
|
|
|
|
def test_format_examples() -> None:
|
|
"""Test format examples."""
|
|
examples = [
|
|
(
|
|
"Add 1 and 2",
|
|
r_(c_(add_, 1, 2)),
|
|
),
|
|
]
|
|
|
|
formatted_examples = format_examples(
|
|
language_name="meow",
|
|
examples=examples,
|
|
ast_printer=AstPrinter(),
|
|
)
|
|
|
|
assert formatted_examples == [
|
|
(
|
|
"Add 1 and 2",
|
|
"```meow\nvar result = add_(1, 2)\n```",
|
|
),
|
|
]
|