From 45d95289ab878c42d0ae87a2c914ee39672b3065 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Dlouh=C3=BD?= <petr.dlouhy@email.cz>
Date: Sun, 14 Apr 2019 18:59:50 +0100
Subject: [PATCH] Python3: fix logging

---
 tests/test_decorators.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/tests/test_decorators.py b/tests/test_decorators.py
index 9f2ae701..54c21f5c 100644
--- a/tests/test_decorators.py
+++ b/tests/test_decorators.py
@@ -20,7 +20,13 @@ Test decorators.
 
 import unittest
 import time
-from io import BytesIO
+
+try:
+    from cStringIO import StringIO
+except ImportError:
+    # Python 3
+    from io import StringIO
+
 import linkcheck.decorators
 
 
@@ -36,10 +42,10 @@ class TestDecorators (unittest.TestCase):
         self.assertEqual(f(), 42)
 
     def test_timeit2 (self):
-        log = BytesIO()
+        log = StringIO()
         @linkcheck.decorators.timed(log=log, limit=0)
         def f ():
             time.sleep(1)
             return 42
         self.assertEqual(f(), 42)
-        self.assertTrue(log.getvalue())
+        self.assertEqual(log.getvalue(), 'f took 1.00 seconds\n()\n{}\n')