Merge pull request astanin#139 from saveshodhan/hf-firstrow-fix

astanin · web-flow · commit adc5630c8cd4 · 2022-06-22T08:39:28.000Z
fix output for empty list with headers="firstrow"
diff --git a/tabulate.py b/tabulate.py
@@ -1253,6 +1253,8 @@ def _normalize_tabular_data(tabular_data, headers, showindex="default"):
             headers = rows[0]
         headers = list(map(str, headers))  # headers should be strings
         rows = rows[1:]
+    elif headers == "firstrow":
+        headers = []
 
     headers = list(map(str, headers))
     rows = list(map(list, rows))
diff --git a/test/test_output.py b/test/test_output.py
@@ -2104,6 +2104,13 @@ def test_empty_data_without_headers():
     assert_equal(expected, result)
 
 
+def test_empty_data_with_headers():
+    "Output: table with empty data and headers as firstrow"
+    expected = ""
+    result = tabulate([], headers="firstrow")
+    assert_equal(expected, result)
+
+
 def test_floatfmt():
     "Output: floating point format"
     result = tabulate([["1.23456789"], [1.0]], floatfmt=".3f", tablefmt="plain")
