mirror of
https://github.com/ihabunek/toot
synced 2024-12-29 02:10:37 +01:00
158 lines
5.8 KiB
Python
158 lines
5.8 KiB
Python
from toot.wcstring import wc_wrap, trunc, pad, fit_text
|
|
|
|
|
|
def test_pad():
|
|
# guitar symbol will occupy two cells, so padded text should be 1
|
|
# character shorter
|
|
text = 'Frank Zappa 🎸'
|
|
|
|
# Negative values are basically ignored
|
|
assert pad(text, -100) is text
|
|
|
|
# Padding to length smaller than text length does nothing
|
|
assert pad(text, 11) is text
|
|
assert pad(text, 12) is text
|
|
assert pad(text, 13) is text
|
|
assert pad(text, 14) is text
|
|
|
|
assert pad(text, 15) == 'Frank Zappa 🎸 '
|
|
assert pad(text, 16) == 'Frank Zappa 🎸 '
|
|
assert pad(text, 17) == 'Frank Zappa 🎸 '
|
|
assert pad(text, 18) == 'Frank Zappa 🎸 '
|
|
assert pad(text, 19) == 'Frank Zappa 🎸 '
|
|
assert pad(text, 20) == 'Frank Zappa 🎸 '
|
|
|
|
|
|
def test_trunc():
|
|
text = 'Frank Zappa 🎸'
|
|
|
|
assert trunc(text, 1) == '…'
|
|
assert trunc(text, 2) == 'F…'
|
|
assert trunc(text, 3) == 'Fr…'
|
|
assert trunc(text, 4) == 'Fra…'
|
|
assert trunc(text, 5) == 'Fran…'
|
|
assert trunc(text, 6) == 'Frank…'
|
|
assert trunc(text, 7) == 'Frank…'
|
|
assert trunc(text, 8) == 'Frank Z…'
|
|
assert trunc(text, 9) == 'Frank Za…'
|
|
assert trunc(text, 10) == 'Frank Zap…'
|
|
assert trunc(text, 11) == 'Frank Zapp…'
|
|
assert trunc(text, 12) == 'Frank Zappa…'
|
|
assert trunc(text, 13) == 'Frank Zappa…'
|
|
|
|
# Truncating to length larger than text length does nothing
|
|
assert trunc(text, 14) is text
|
|
assert trunc(text, 15) is text
|
|
assert trunc(text, 16) is text
|
|
assert trunc(text, 17) is text
|
|
assert trunc(text, 18) is text
|
|
assert trunc(text, 19) is text
|
|
assert trunc(text, 20) is text
|
|
|
|
|
|
def test_fit_text():
|
|
text = 'Frank Zappa 🎸'
|
|
|
|
assert fit_text(text, 1) == '…'
|
|
assert fit_text(text, 2) == 'F…'
|
|
assert fit_text(text, 3) == 'Fr…'
|
|
assert fit_text(text, 4) == 'Fra…'
|
|
assert fit_text(text, 5) == 'Fran…'
|
|
assert fit_text(text, 6) == 'Frank…'
|
|
assert fit_text(text, 7) == 'Frank…'
|
|
assert fit_text(text, 8) == 'Frank Z…'
|
|
assert fit_text(text, 9) == 'Frank Za…'
|
|
assert fit_text(text, 10) == 'Frank Zap…'
|
|
assert fit_text(text, 11) == 'Frank Zapp…'
|
|
assert fit_text(text, 12) == 'Frank Zappa…'
|
|
assert fit_text(text, 13) == 'Frank Zappa…'
|
|
assert fit_text(text, 14) == 'Frank Zappa 🎸'
|
|
assert fit_text(text, 15) == 'Frank Zappa 🎸 '
|
|
assert fit_text(text, 16) == 'Frank Zappa 🎸 '
|
|
assert fit_text(text, 17) == 'Frank Zappa 🎸 '
|
|
assert fit_text(text, 18) == 'Frank Zappa 🎸 '
|
|
assert fit_text(text, 19) == 'Frank Zappa 🎸 '
|
|
assert fit_text(text, 20) == 'Frank Zappa 🎸 '
|
|
|
|
|
|
def test_wc_wrap_plain_text():
|
|
lorem = (
|
|
"Eius voluptas eos praesentium et tempore. Quaerat nihil voluptatem "
|
|
"excepturi reiciendis sapiente voluptate natus. Tenetur occaecati "
|
|
"velit dicta dolores. Illo reiciendis nulla ea. Facilis nostrum non "
|
|
"qui inventore sit."
|
|
)
|
|
|
|
assert list(wc_wrap(lorem, 50)) == [
|
|
#01234567890123456789012345678901234567890123456789 # noqa
|
|
"Eius voluptas eos praesentium et tempore. Quaerat",
|
|
"nihil voluptatem excepturi reiciendis sapiente",
|
|
"voluptate natus. Tenetur occaecati velit dicta",
|
|
"dolores. Illo reiciendis nulla ea. Facilis nostrum",
|
|
"non qui inventore sit.",
|
|
]
|
|
|
|
|
|
def test_wc_wrap_plain_text_wrap_on_any_whitespace():
|
|
lorem = (
|
|
"Eius\t\tvoluptas\teos\tpraesentium\tet\ttempore.\tQuaerat\tnihil\tvoluptatem\t"
|
|
"excepturi\nreiciendis\n\nsapiente\nvoluptate\nnatus.\nTenetur\noccaecati\n"
|
|
"velit\rdicta\rdolores.\rIllo\rreiciendis\rnulla\r\r\rea.\rFacilis\rnostrum\rnon\r"
|
|
"qui\u2003inventore\u2003\u2003sit." # em space
|
|
)
|
|
|
|
assert list(wc_wrap(lorem, 50)) == [
|
|
#01234567890123456789012345678901234567890123456789 # noqa
|
|
"Eius voluptas eos praesentium et tempore. Quaerat",
|
|
"nihil voluptatem excepturi reiciendis sapiente",
|
|
"voluptate natus. Tenetur occaecati velit dicta",
|
|
"dolores. Illo reiciendis nulla ea. Facilis nostrum",
|
|
"non qui inventore sit.",
|
|
]
|
|
|
|
|
|
def test_wc_wrap_text_with_wide_chars():
|
|
lorem = (
|
|
"☕☕☕☕☕ voluptas eos praesentium et 🎸🎸🎸🎸🎸. Quaerat nihil "
|
|
"voluptatem excepturi reiciendis sapiente voluptate natus."
|
|
)
|
|
|
|
assert list(wc_wrap(lorem, 50)) == [
|
|
#01234567890123456789012345678901234567890123456789 # noqa
|
|
"☕☕☕☕☕ voluptas eos praesentium et 🎸🎸🎸🎸🎸.",
|
|
"Quaerat nihil voluptatem excepturi reiciendis",
|
|
"sapiente voluptate natus.",
|
|
]
|
|
|
|
|
|
def test_wc_wrap_hard_wrap():
|
|
lorem = (
|
|
"☕☕☕☕☕voluptaseospraesentiumet🎸🎸🎸🎸🎸.Quaeratnihil"
|
|
"voluptatemexcepturireiciendissapientevoluptatenatus."
|
|
)
|
|
|
|
assert list(wc_wrap(lorem, 50)) == [
|
|
#01234567890123456789012345678901234567890123456789 # noqa
|
|
"☕☕☕☕☕voluptaseospraesentiumet🎸🎸🎸🎸🎸.Quaer",
|
|
"atnihilvoluptatemexcepturireiciendissapientevolupt",
|
|
"atenatus.",
|
|
]
|
|
|
|
|
|
def test_wc_wrap_indented():
|
|
lorem = (
|
|
" Eius voluptas eos praesentium et tempore. Quaerat nihil voluptatem "
|
|
" excepturi reiciendis sapiente voluptate natus. Tenetur occaecati "
|
|
" velit dicta dolores. Illo reiciendis nulla ea. Facilis nostrum non "
|
|
" qui inventore sit."
|
|
)
|
|
|
|
assert list(wc_wrap(lorem, 50)) == [
|
|
#01234567890123456789012345678901234567890123456789 # noqa
|
|
"Eius voluptas eos praesentium et tempore. Quaerat",
|
|
"nihil voluptatem excepturi reiciendis sapiente",
|
|
"voluptate natus. Tenetur occaecati velit dicta",
|
|
"dolores. Illo reiciendis nulla ea. Facilis nostrum",
|
|
"non qui inventore sit.",
|
|
]
|