1 files changed, 107 insertions, 0 deletions
diff --git a/tests/test_width_and_alignment/test_column_max_width.py b/tests/test_width_and_alignment/test_column_max_width.py
new file mode 100644
index 0000000..696c9bf
--- /dev/null
+++ b/tests/test_width_and_alignment/test_column_max_width.py
@@ -0,0 +1,107 @@
+"""Test function in module."""
+
+import pytest
+
+from terminaltables.width_and_alignment import column_max_width, max_dimensions
+
+
+@pytest.fixture(autouse=True)
+def patch(monkeypatch):
+    """Monkeypatch before every test function in this module.
+
+    :param monkeypatch: pytest fixture.
+    """
+    monkeypatch.setattr('terminaltables.width_and_alignment.terminal_size', lambda: (79, 24))
+
+
+def test_empty():
+    """Test with zero-length cells."""
+    assert column_max_width(max_dimensions([['']])[0], 0, 0, 0, 0) == 79
+    assert column_max_width(max_dimensions([['', '', '']])[0], 0, 0, 0, 0) == 79
+    assert column_max_width(max_dimensions([['', '', ''], ['', '', '']])[0], 0, 0, 0, 0) == 79
+
+    assert column_max_width(max_dimensions([['']])[0], 0, 2, 1, 2) == 75
+    assert column_max_width(max_dimensions([['', '', '']])[0], 0, 2, 1, 2) == 69
+    assert column_max_width(max_dimensions([['', '', ''], ['', '', '']])[0], 0, 2, 1, 2) == 69
+
+
+def test_single_line():
+    """Test with single-line cells."""
+    table_data = [
+        ['Name', 'Color', 'Type'],
+        ['Avocado', 'green', 'nut'],
+        ['Tomato', 'red', 'fruit'],
+        ['Lettuce', 'green', 'vegetable'],
+    ]
+    inner_widths = max_dimensions(table_data)[0]
+
+    # '| Lettuce | green | vegetable |'
+    outer, inner, padding = 2, 1, 2
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 55
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 53
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 57
+
+    # ' Lettuce | green | vegetable '
+    outer = 0
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 57
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 55
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 59
+
+    # '| Lettuce  green  vegetable |'
+    outer, inner = 2, 0
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 57
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 55
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 59
+
+    # ' Lettuce  green  vegetable '
+    outer = 0
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 59
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 57
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 61
+
+    # '|Lettuce |green |vegetable |'
+    outer, inner, padding = 2, 1, 1
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 58
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 56
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 60
+
+    # '|Lettuce     |green     |vegetable     |'
+    padding = 5
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 46
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 44
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 48
+
+    table_data = [
+        ['Name', 'Color', 'Type'],
+        ['Avocado', 'green', 'nut'],
+        ['Tomato', 'red', 'fruit'],
+        ['Lettuce', 'green', 'vegetable'],
+        ['Watermelon', 'green', 'fruit'],
+    ]
+    inner_widths = max_dimensions(table_data)[0]
+    outer, inner, padding = 2, 1, 2
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 55
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 50
+    assert column_max_width(inner_widths, 2, outer, inner, padding) == 54
+
+
+def test_multi_line(monkeypatch):
+    """Test with multi-line cells.
+
+    :param monkeypatch: pytest fixture.
+    """
+    table_data = [
+        ['Show', 'Characters'],
+        ['Rugrats', ('Tommy Pickles, Chuckie Finster, Phillip DeVille, Lillian DeVille, Angelica Pickles,\n'
+                     'Susie Carmichael, Dil Pickles, Kimi Finster, Spike')],
+        ['South Park', 'Stan Marsh, Kyle Broflovski, Eric Cartman, Kenny McCormick']
+    ]
+    inner_widths = max_dimensions(table_data)[0]
+    outer, inner, padding = 2, 1, 2
+
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == -11
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 62
+
+    monkeypatch.setattr('terminaltables.width_and_alignment.terminal_size', lambda: (100, 24))
+    assert column_max_width(inner_widths, 0, outer, inner, padding) == 10
+    assert column_max_width(inner_widths, 1, outer, inner, padding) == 83