Fix tests for multiline TerminalStatus support

1 files changed, 9 insertions, 2 deletions

diff --git a/ttystatus/status.py b/ttystatus/status.py
index 61adcbe..0f57f74 100644
--- a/ttystatus/status.py
+++ b/ttystatus/status.py
@@ -40,7 +40,7 @@ class TerminalStatus(object):
         '''Add a new widget to the status display.'''
         self._widget_rows[-1].append(widget)
 
-    def start_new_line(self):
+    def start_new_line(self):  # pragma: no cover
         '''Start a new line of widgets.'''
         self._widget_rows.append([])
 
@@ -55,11 +55,18 @@ class TerminalStatus(object):
         '''
 
         for i, line in enumerate(format_string.split('\n')):
-            if i > 0:
+            if i > 0:  # pragma: no cover
                 self.start_new_line()
             for widget in ttystatus.parse(line):
                 self.add(widget)
 
+    @property
+    def widgets(self):
+        result = []
+        for row in self._widget_rows:
+            result += row
+        return result
+
     def clear(self):
         '''Remove all widgets.'''
         self._widget_rows = [[]]