fix test

methane · methane · commit 9888906a6da3 · 2016-01-10T14:53:45.000+09:00
diff --git a/pymysql/tests/test_connection.py b/pymysql/tests/test_connection.py
@@ -1,10 +1,10 @@
 import datetime
-import decimal
 import sys
 import time
 import unittest2
 import pymysql
 from pymysql.tests import base
+from pymysql._compat import text_type
 
 
 class TempUser:
@@ -534,7 +534,7 @@ def test_escape_fallback_encoder(self):
         class Custom(str):
             pass
 
-        mapping = {pymysql.text_type: pymysql.escape_string}
+        mapping = {text_type: pymysql.escape_string}
         self.assertEqual(con.escape(Custom('foobar'), mapping), "'foobar'")
 
     def test_escape_no_default(self):
