1 files changed, 104 insertions, 0 deletions
diff --git a/tests/test_build/test_build_row.py b/tests/test_build/test_build_row.py
new file mode 100644
index 0000000..ce55944
--- /dev/null
+++ b/tests/test_build/test_build_row.py
@@ -0,0 +1,104 @@
+"""Test function in module."""
+
+from terminaltables.build import build_row
+
+
+def test_one_line():
+    """Test with one line cells."""
+    row = [
+        ['Left Cell'], ['Center Cell'], ['Right Cell'],
+    ]
+    actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
+    expected = [
+        ('>', 'Left Cell', '|', 'Center Cell', '|', 'Right Cell', '<'),
+    ]
+    assert actual == expected
+
+
+def test_two_line():
+    """Test with two line cells."""
+    row = [
+        [
+            'Left ',
+            'Cell1',
+        ],
+
+        [
+            'Center',
+            'Cell2 ',
+        ],
+
+        [
+            'Right',
+            'Cell3',
+        ],
+    ]
+    actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
+    expected = [
+        ('>', 'Left ', '|', 'Center', '|', 'Right', '<'),
+        ('>', 'Cell1', '|', 'Cell2 ', '|', 'Cell3', '<'),
+    ]
+    assert actual == expected
+
+
+def test_three_line():
+    """Test with three line cells."""
+    row = [
+        [
+            'Left ',
+            'Cell1',
+            '     ',
+        ],
+
+        [
+            'Center',
+            'Cell2 ',
+            '      ',
+        ],
+
+        [
+            'Right',
+            'Cell3',
+            '     ',
+        ],
+    ]
+    actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
+    expected = [
+        ('>', 'Left ', '|', 'Center', '|', 'Right', '<'),
+        ('>', 'Cell1', '|', 'Cell2 ', '|', 'Cell3', '<'),
+        ('>', '     ', '|', '      ', '|', '     ', '<'),
+    ]
+    assert actual == expected
+
+
+def test_single():
+    """Test with single cell."""
+    actual = [tuple(i) for i in build_row([['Cell']], '>', '|', '<')]
+    expected = [
+        ('>', 'Cell', '<'),
+    ]
+    assert actual == expected
+
+
+def test_empty():
+    """Test with empty cell."""
+    actual = [tuple(i) for i in build_row([['']], '>', '|', '<')]
+    expected = [
+        ('>', '', '<'),
+    ]
+    assert actual == expected
+
+
+def test_no_cells():
+    """Test with no cells."""
+    actual = [tuple(i) for i in build_row([[]], '>', '|', '<')]
+    expected = [
+        ('>', '<'),
+    ]
+    assert actual == expected
+
+    actual = [tuple(i) for i in build_row([], '>', '|', '<')]
+    expected = [
+        ('>', '<'),
+    ]
+    assert actual == expected