1 files changed, 11 insertions, 4 deletions
diff --git a/tests/test_transports.py b/tests/test_transports.py
index 3b6e3d6..42f7729 100644
--- a/tests/test_transports.py
+++ b/tests/test_transports.py
@@ -1,13 +1,20 @@
 """Tests for transports.py."""
 
 import unittest
-from unittest import mock
 
-import asyncio
-from asyncio import transports
+import trollius as asyncio
+from trollius import test_utils
+from trollius import transports
+from trollius.test_utils import mock
 
+try:
+    memoryview
+except NameError:
+    # Python 2.6
+    memoryview = buffer
 
-class TransportTests(unittest.TestCase):
+
+class TransportTests(test_utils.TestCase):
 
     def test_ctor_extra_is_none(self):
         transport = asyncio.Transport()