Minor fixes and linter

Author: gvbgduh

tests/test_connection_options.py

@@ -56,7 +56,9 @@ def test_mysql_explicit_ssl():
 
 
 def test_aiopg_pool_size():
-    backend = AiopgBackend("postgresql+aiopg://localhost/database?min_size=1&max_size=20")
+    backend = AiopgBackend(
+        "postgresql+aiopg://localhost/database?min_size=1&max_size=20"
+    )
     kwargs = backend._get_connection_kwargs()
     assert kwargs == {"minsize": 1, "maxsize": 20}
 

tests/test_databases.py

@@ -77,9 +77,9 @@ def create_test_database():
     # Create test databases with tables creation
     for url in DATABASE_URLS:
         database_url = DatabaseURL(url)
-        if database_url.dialect == "mysql":
+        if database_url.scheme == "mysql":
             url = str(database_url.replace(driver="pymysql"))
-        elif database_url.driver == "aiopg":
+        elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.create_all(engine)
@@ -90,9 +90,9 @@ def create_test_database():
     # Drop test databases
     for url in DATABASE_URLS:
         database_url = DatabaseURL(url)
-        if database_url.dialect == "mysql":
+        if database_url.scheme == "mysql":
             url = str(database_url.replace(driver="pymysql"))
-        elif database_url.driver == "aiopg":
+        elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.drop_all(engine)

tests/test_integration.py

@@ -28,9 +28,9 @@ def create_test_database():
     # Create test databases
     for url in DATABASE_URLS:
         database_url = DatabaseURL(url)
-        if database_url.dialect == "mysql":
+        if database_url.scheme == "mysql":
             url = str(database_url.replace(driver="pymysql"))
-        elif database_url.driver == "aiopg":
+        elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.create_all(engine)
@@ -41,9 +41,9 @@ def create_test_database():
     # Drop test databases
     for url in DATABASE_URLS:
         database_url = DatabaseURL(url)
-        if database_url.dialect == "mysql":
+        if database_url.scheme == "mysql":
             url = str(database_url.replace(driver="pymysql"))
-        elif database_url.driver == "aiopg":
+        elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.drop_all(engine)