Fix CI

aiocassandra/__init__.py

@@ -1,4 +1,4 @@
-from aiocassandra.session import aiosession
 from aiocassandra.cqlengine.models import AioModel
+from aiocassandra.session import aiosession
 
 __all__ = ['aiosession', 'AioModel']

aiocassandra/cqlengine/query.py

@@ -3,13 +3,12 @@
 """
 from warnings import warn
 
-from cassandra.cqlengine.query import (DMLQuery, ModelQuerySet, check_applied,
-                                       SimpleStatement, conn, ValidationError,
-                                       EqualsOperator)
-from cassandra.cqlengine import CQLEngineException
-from cassandra.cqlengine import columns
-from cassandra.cqlengine.statements import (UpdateStatement, DeleteStatement,
-                                            BaseCQLStatement, InsertStatement)
+from cassandra.cqlengine import CQLEngineException, columns
+from cassandra.cqlengine.query import (DMLQuery, EqualsOperator, ModelQuerySet,
+                                       SimpleStatement, ValidationError,
+                                       check_applied, conn)
+from cassandra.cqlengine.statements import (BaseCQLStatement, DeleteStatement,
+                                            InsertStatement, UpdateStatement)
 
 
 async def _execute_statement(model,
@@ -325,7 +324,8 @@ async def async_update(self, **values):
             if col_op == 'remove' and isinstance(col, columns.Map):
                 if not isinstance(val, set):
                     raise ValidationError(
-                        "Cannot apply update operation '{0}' on column '{1}' with value '{2}'. A set is required."
+                        "Cannot apply update operation '{0}' on column '{1}' "
+                        "with value '{2}'. A set is required."
                         .format(col_op, col_name, val))
                 val = {v: None for v in val}
             else:

tests/conftest.py

@@ -4,8 +4,8 @@
 
 import pytest
 from cassandra.cluster import Cluster
-from cassandra.cqlengine import management
 from cassandra.cqlengine import connection as cqlengine_connection
+from cassandra.cqlengine import management
 
 from aiocassandra import aiosession
 

tests/test_async_cqlengine.py

@@ -21,7 +21,9 @@ async def test_queryset_async_functions(cqlengine_management):
     """
     cqlengine_management.sync_table(User)
 
-    # test: Model.objects.async_create(), Model.objects.async_all(), Model.objects.async_get()
+    # test: Model.objects.async_create()
+    #       Model.objects.async_all()
+    #       Model.objects.async_get()
     username1 = 'test-username-1'
     await User.objects.async_create(user_id=uuid.uuid4(), username=username1)
     users = await User.objects.async_all()

tox.ini

@@ -9,8 +9,8 @@ deps =
     flake8
     isort
 commands =
-    flake8 --show-source aiocassandra.py setup.py
-    isort --check-only aiocassandra.py setup.py
+    flake8 --show-source aiocassandra
+    isort --check-only -rc aiocassandra
     flake8 --show-source tests
     isort --check-only -rc tests --diff