Fix mysqlx connection and style tests

nmariz · nmariz · commit ce5b352c8033 · 2017-12-05T10:40:41.000Z
diff --git a/lib/mysqlx/statement.py b/lib/mysqlx/statement.py
@@ -23,15 +23,13 @@
 
 """Implementation of Statements."""
 
-import copy
 import json
-import re
 
 from .errors import ProgrammingError
 from .expr import ExprParser
 from .compat import STRING_TYPES
 from .dbdoc import DbDoc
-from .result import SqlResult, Result, ColumnType
+from .result import SqlResult, Result
 from .protobuf import mysqlxpb_enum
 
 
diff --git a/tests/test_mysqlx_connection.py b/tests/test_mysqlx_connection.py
@@ -38,6 +38,9 @@
 else:
     from urllib import quote_plus
 
+from .test_mysqlx_crud import drop_table
+
+
 LOGGER = logging.getLogger(tests.LOGGER_NAME)
 
 _URI_TEST_RESULTS = (  # (uri, result)
@@ -420,7 +423,7 @@ def test_rollback(self):
         self.session.rollback()
         self.assertEqual(table.count(), 0)
 
-        schema.drop_table(table_name)
+        drop_table(schema, table_name)
 
     def test_commit(self):
         table_name = "t2"
@@ -441,7 +444,7 @@ def test_commit(self):
         self.session.commit()
         self.assertEqual(table.count(), 1)
 
-        schema.drop_table(table_name)
+        drop_table(schema, table_name)
 
     def test_close(self):
         session = mysqlx.get_session(self.connect_kwargs)
