Merge pull request #11 from python-tableformatter/unit_tests

Added more unit tests related to text wrapping

Author: tleonhardt

tests/test_utils.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 """
-Unit testing of tableformatter utility functtions.
+Unit testing of tableformatter utility functions.
 """
 import pytest
 
@@ -11,15 +11,26 @@
 def empty_text():
     return ''
 
-
 @pytest.fixture
 def sample_text():
     return 'FooBar'
 
+@pytest.fixture
+def long_text():
+    return 'FooBarBaz'
+
 @pytest.fixture
 def tabbed_text():
     return 'Foo\tBar'
 
+@pytest.fixture
+def multiline_text():
+    return 'FooBar\nBaz'
+
+@pytest.fixture
+def whitespace_text():
+    return 'FooBar  \n    '
+
 
 # Test text wrapping
 def test_empty_wrap(empty_text):
@@ -37,6 +48,26 @@ def test_simple_wrap(sample_text):
     wrapped = tf._text_wrap(sample_text, width=3)
     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_multiline_no_wrap(multiline_text):
+    expected = ['FooBar Baz']
+    wrapped = tf._text_wrap(multiline_text)
+    assert expected == wrapped
+
+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
+
 
 # Test tab translation and printable width calculation
 def test_empty_translation(empty_text):