Update test_format.py

twaugh · Oct 6, 2015 · 85d4593 · 85d4593
1 parent 9ec94d7
commit 85d4593
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_format.py b/tests/test_format.py
@@ -32,7 +32,7 @@ def test_timestamp(self):
         # Should output in local time
         expected = 'Jan 01 01:00:00'
 
-        assert expected in formatter.format([entry])
+        assert expected in formatter.format(entry)
 
     @pytest.mark.parametrize(('entry', 'expected'), [
         ({'MESSAGE': 'message'},
@@ -69,7 +69,7 @@ def test_format(self, entry, expected):
         entry['__REALTIME_TIMESTAMP'] = datetime.fromtimestamp(0,
                                                                tz=timezone.utc)
         formatter = get_formatter('short')
-        formatted = formatter.format([entry])
+        formatted = formatter.format(entry)
         date = 'Jan 01 00:00:00 '
         assert formatted.startswith(date)
         assert formatted[len(date):] == expected + '\n'