mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-07 20:00:59 +00:00
Junyan Qin
18 lines
442 B
Python
18 lines
442 B
Python
from unittest.mock import Mock
|
|
|
|
from langbot.pkg.pipeline.longtext.strategies.image import Text2ImageStrategy
|
|
|
|
|
|
class _WideFont:
|
|
def getlength(self, text: str) -> int:
|
|
return len(text) * 100
|
|
|
|
|
|
def test_image_strategy_line_split_always_consumes_input():
|
|
strategy = Text2ImageStrategy(Mock())
|
|
|
|
lines = strategy._split_text_lines('abc', 1, _WideFont())
|
|
|
|
assert lines == ['a', 'b', 'c']
|
|
assert ''.join(lines) == 'abc'
|