28 lines
798 B
Python
28 lines
798 B
Python
import tempfile
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from haiku.rag.config import Config
|
|
from haiku.rag.converters import get_converter
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_code_file_wrapped_in_code_block():
|
|
"""Test that code files are wrapped in markdown code blocks."""
|
|
python_code = '''def hello_world():
|
|
print("Hello, World!")
|
|
return "success"'''
|
|
|
|
with tempfile.NamedTemporaryFile(mode="w", suffix=".py") as f:
|
|
f.write(python_code)
|
|
f.flush()
|
|
temp_path = Path(f.name)
|
|
|
|
converter = get_converter(Config)
|
|
document = await converter.convert_file(temp_path)
|
|
result = document.export_to_markdown()
|
|
|
|
assert result.startswith("```\n")
|
|
assert result.endswith("\n```")
|
|
assert "def hello_world():" in result
|