god dammit

Author: GiovanniMCMXCIX

tests/test_get_all_catalog.py

@@ -24,17 +24,22 @@
 SOFTWARE.
 """
 
-import unittest
-import asyncio
-import uvloop
 import async_connect as connect
+import unittest
+import sys
 
 
 class TestGetAllCatalog(unittest.TestCase):
     def setUp(self):
-        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
-        self.loop = asyncio.get_event_loop()
-        self.connect = connect.Client(loop=self.loop)
+        if sys.version_info[1] == 6:
+            import asyncio
+            import uvloop
+            asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+            self.loop = asyncio.get_event_loop()
+            self.connect = connect.Client(loop=self.loop)
+        else:
+            self.connect = connect.Client()
+            self.loop = self.connect.loop
 
     def test_release(self):
         async def test():

tests/test_get_catalog.py

@@ -24,17 +24,22 @@
 SOFTWARE.
 """
 
-import unittest
-import asyncio
-import uvloop
 import async_connect as connect
+import unittest
+import sys
 
 
 class TestGetAllCatalog(unittest.TestCase):
     def setUp(self):
-        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
-        self.loop = asyncio.get_event_loop()
-        self.connect = connect.Client(loop=self.loop)
+        if sys.version_info[1] == 6:
+            import asyncio
+            import uvloop
+            asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+            self.loop = asyncio.get_event_loop()
+            self.connect = connect.Client(loop=self.loop)
+        else:
+            self.connect = connect.Client()
+            self.loop = self.connect.loop
 
     def test_release(self):
         async def test():

tests/test_search.py

@@ -24,17 +24,22 @@
 SOFTWARE.
 """
 
-import unittest
-import asyncio
-import uvloop
 import async_connect as connect
+import unittest
+import sys
 
 
 class TestSearch(unittest.TestCase):
     def setUp(self):
-        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
-        self.loop = asyncio.get_event_loop()
-        self.connect = connect.Client(loop=self.loop)
+        if sys.version_info[1] == 6:
+            import asyncio
+            import uvloop
+            asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+            self.loop = asyncio.get_event_loop()
+            self.connect = connect.Client(loop=self.loop)
+        else:
+            self.connect = connect.Client()
+            self.loop = self.connect.loop
 
     def test_release(self):
         async def test():