diff --git a/tests/test_reader.py b/tests/test_reader.py
new file mode 100644
index 00000000..8b7803dc
--- /dev/null
+++ b/tests/test_reader.py
@@ -0,0 +1,22 @@
+import tempfile
+from pathlib import Path
+
+from haiku.rag.reader import FileReader
+
+
+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)
+
+        result = FileReader.parse_file(temp_path)
+
+        assert result.startswith("```python\n")
+        assert result.endswith("\n```")
+        assert "def hello_world():" in result