fix: Remove pycon output lines when rendering source as console

src/markdown_exec/formatters/pycon.py

@@ -16,11 +16,13 @@
 
 def _transform_source(code: str) -> tuple[str, str]:
     python_lines = []
+    pycon_lines = []
     for line in code.split("\n"):
         if line.startswith((">>> ", "... ")):
+            pycon_lines.append(line)
             python_lines.append(line[4:])
     python_code = "\n".join(python_lines)
-    return python_code, code
+    return python_code, "\n".join(pycon_lines)
 
 
 def _format_pycon(**kwargs: Any) -> Markup:

tests/test_python.py

@@ -162,3 +162,23 @@ def fraise():
     assert "strawberry" in html
     assert "fraise()" in caplog.text
     assert 'raise RuntimeError("strawberry")' in caplog.text
+
+
+def test_removing_output_from_pycon_code(md: Markdown) -> None:
+    """Assert output lines are removed from pycon snippets.
+
+    Parameters:
+        md: A Markdown instance (fixture).
+    """
+    html = md.convert(
+        dedent(
+            """
+            ```pycon exec="1" source="console"
+            >>> print("ok")
+            ko
+            ```
+            """,
+        ),
+    )
+    assert "ok" in html
+    assert "ko" not in html