Handle special case of having an empty table.

Fixes #1
mailund · Sep 30, 2016 · b3b57b8 · b3b57b8
1 parent 34e6890
commit b3b57b8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/statusbar/__init__.py b/statusbar/__init__.py
@@ -277,6 +277,10 @@ def format_table(self, width=None,
         aligned across lines and then returns formatted lines as a list of
         strings.
         """
+        # handle the special case of an empty table.
+        if len(self._lines) == 0:
+            return []
+
         if width is None:  # pragma: no cover
             width = shutil.get_terminal_size()[0]
 

diff --git a/tests/__init__.py b/tests/__init__.py
@@ -251,6 +251,7 @@ def test_field_width_calculations(self):
 
     def test_table_formatting(self):
         st = statusbar.StatusTable()
+        self.assertEqual(st.format_table(), [])
 
         label1 = "Test"
         sb = st.add_status_line(label1)