def test_multiline_with_wrap(multiline_text):
expected = ['Foo', 'Bar', 'Baz']
wrapped = tf._text_wrap(multiline_text, width=3)
assert expected == wrapped
def test_trailing_whitespace_wrap(whitespace_text):
expected = ['FooBar']
wrapped = tf._text_wrap(whitespace_text)
assert expected == wrapped
def test_multiline_no_wrap(multiline_text):
expected = ['FooBar Baz']
wrapped = tf._text_wrap(multiline_text)
assert expected == wrapped
def test_double_wrap(long_text):
expected = ['Foo', 'Bar', 'Baz']
wrapped = tf._text_wrap(long_text, width=3)
assert expected == wrapped
def test_simple_wrap(sample_text):
expected = ['Foo', 'Bar']
wrapped = tf._text_wrap(sample_text, width=3)
assert expected == wrapped
def test_no_wrap(sample_text):
expected = [sample_text]
wrapped = tf._text_wrap(sample_text)
assert expected == wrapped
def test_empty_wrap(empty_text):
expected = []
wrapped = tf._text_wrap(empty_text)
assert expected == wrapped