- more tests, move some tests out of test_reflection, test_query

1 files changed, 6 insertions, 8 deletions

diff --git a/lib/sqlalchemy/testing/assertsql.py b/lib/sqlalchemy/testing/assertsql.py
index 897f4b3b1..08ee55d57 100644
--- a/lib/sqlalchemy/testing/assertsql.py
+++ b/lib/sqlalchemy/testing/assertsql.py
@@ -1,8 +1,6 @@
 
-from sqlalchemy.interfaces import ConnectionProxy
-from sqlalchemy.engine.default import DefaultDialect
-from sqlalchemy.engine.base import Connection
-from sqlalchemy import util
+from ..engine.default import DefaultDialect
+from .. import util
 import re
 
 class AssertRule(object):
@@ -262,16 +260,16 @@ def _process_assertion_statement(query, context):
     paramstyle = context.dialect.paramstyle
     if paramstyle == 'named':
         pass
-    elif paramstyle =='pyformat':
+    elif paramstyle == 'pyformat':
         query = re.sub(r':([\w_]+)', r"%(\1)s", query)
     else:
         # positional params
         repl = None
-        if paramstyle=='qmark':
+        if paramstyle == 'qmark':
             repl = "?"
-        elif paramstyle=='format':
+        elif paramstyle == 'format':
             repl = r"%s"
-        elif paramstyle=='numeric':
+        elif paramstyle == 'numeric':
             repl = None
         query = re.sub(r':([\w_]+)', repl, query)