summaryrefslogtreecommitdiffstats
path: root/tests/test_all_tables_e2e/test_github_table.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_all_tables_e2e/test_github_table.py')
-rw-r--r--tests/test_all_tables_e2e/test_github_table.py77
1 files changed, 77 insertions, 0 deletions
diff --git a/tests/test_all_tables_e2e/test_github_table.py b/tests/test_all_tables_e2e/test_github_table.py
new file mode 100644
index 0000000..6176215
--- /dev/null
+++ b/tests/test_all_tables_e2e/test_github_table.py
@@ -0,0 +1,77 @@
+"""GithubFlavoredMarkdownTable end to end testing."""
+
+from terminaltables import GithubFlavoredMarkdownTable
+
+
+def test_single_line():
+ """Test single-lined cells."""
+ table_data = [
+ ['Name', 'Color', 'Type'],
+ ['Avocado', 'green', 'nut'],
+ ['Tomato', 'red', 'fruit'],
+ ['Lettuce', 'green', 'vegetable'],
+ ['Watermelon', 'green'],
+ [],
+ ]
+ table = GithubFlavoredMarkdownTable(table_data)
+ table.inner_footing_row_border = True
+ table.justify_columns[0] = 'left'
+ table.justify_columns[1] = 'center'
+ table.justify_columns[2] = 'right'
+ actual = table.table
+
+ expected = (
+ '| Name | Color | Type |\n'
+ '|:-----------|:-----:|----------:|\n'
+ '| Avocado | green | nut |\n'
+ '| Tomato | red | fruit |\n'
+ '| Lettuce | green | vegetable |\n'
+ '| Watermelon | green | |\n'
+ '| | | |'
+ )
+ assert actual == expected
+
+
+def test_multi_line():
+ """Test multi-lined cells."""
+ table_data = [
+ ['Show', 'Characters'],
+ ['Rugrats', 'Tommy Pickles, Chuckie Finster, Phillip DeVille, Lillian DeVille, Angelica Pickles,\nDil Pickles'],
+ ['South Park', 'Stan Marsh, Kyle Broflovski, Eric Cartman, Kenny McCormick']
+ ]
+ table = GithubFlavoredMarkdownTable(table_data)
+
+ # Test defaults.
+ actual = table.table
+ expected = (
+ '| Show | Characters |\n'
+ '|------------|-------------------------------------------------------------------------------------|\n'
+ '| Rugrats | Tommy Pickles, Chuckie Finster, Phillip DeVille, Lillian DeVille, Angelica Pickles, |\n'
+ '| | Dil Pickles |\n'
+ '| South Park | Stan Marsh, Kyle Broflovski, Eric Cartman, Kenny McCormick |'
+ )
+ assert actual == expected
+
+ # Test inner row border.
+ table.inner_row_border = True
+ actual = table.table
+ expected = (
+ '| Show | Characters |\n'
+ '|------------|-------------------------------------------------------------------------------------|\n'
+ '| Rugrats | Tommy Pickles, Chuckie Finster, Phillip DeVille, Lillian DeVille, Angelica Pickles, |\n'
+ '| | Dil Pickles |\n'
+ '| South Park | Stan Marsh, Kyle Broflovski, Eric Cartman, Kenny McCormick |'
+ )
+ assert actual == expected
+
+ # Justify right.
+ table.justify_columns = {1: 'right'}
+ actual = table.table
+ expected = (
+ '| Show | Characters |\n'
+ '|------------|------------------------------------------------------------------------------------:|\n'
+ '| Rugrats | Tommy Pickles, Chuckie Finster, Phillip DeVille, Lillian DeVille, Angelica Pickles, |\n'
+ '| | Dil Pickles |\n'
+ '| South Park | Stan Marsh, Kyle Broflovski, Eric Cartman, Kenny McCormick |'
+ )
+ assert actual == expected