Add a test written by darikg. Not actually executed, please confirm it.

1 files changed, 14 insertions, 0 deletions

diff --git a/tests/test_tabulate.py b/tests/test_tabulate.py
new file mode 100644
index 00000000..528255d1
--- /dev/null
+++ b/tests/test_tabulate.py
@@ -0,0 +1,14 @@
+from pgcli.packages.tabulate import tabulate
+from textwrap import dedent
+
+
+def test_dont_strip_leading_whitespace():
+    data = [['    abc']]
+    headers = ['xyz']
+    tbl, _ = tabulate(data, headers, tablefmt='psql')
+    assert tbl == dedent('''
+        +---------+
+        | xyz     |
+        |---------|
+        |     abc |
+        +---------+ ''').strip()