30 lines
929 B
Python
30 lines
929 B
Python
"""
|
|
Tests for escape_code_fences_for_display.
|
|
|
|
B1: Escape triple-backtick markers inside reasoning text before wrapping
|
|
in an outer ``` fence, so inner ``` doesn't break the outer block.
|
|
"""
|
|
|
|
import pytest
|
|
from gateway.stream_consumer import escape_code_fences_for_display
|
|
|
|
|
|
class TestEscapeCodeFencesForDisplay:
|
|
"""escape_code_fences_for_display prevents inner ``` from breaking
|
|
the outer code block used to render reasoning."""
|
|
|
|
|
|
def test_single_fence_escaped(self):
|
|
text = "model used ```python\nx = 1\n``` in its thinking"
|
|
result = escape_code_fences_for_display(text)
|
|
assert "```" not in result
|
|
assert "\\`\\`\\`" in result
|
|
|
|
def test_multiple_fences_all_escaped(self):
|
|
text = "```\nblock1\n``` and ```python\nblock2\n```"
|
|
result = escape_code_fences_for_display(text)
|
|
assert result.count("```") == 0
|
|
assert result.count("\\`\\`\\`") == 4
|
|
|
|
|