Add frozenset to escape_sequence types

tonal · tonal · commit 88ca33c6e066 · 2016-05-19T17:13:12.000+06:00
diff --git a/pymysql/converters.py b/pymysql/converters.py
@@ -355,6 +355,7 @@ def convert_characters(connection, field, data):
     tuple: escape_sequence,
     list: escape_sequence,
     set: escape_sequence,
+    frozenset: escape_sequence,
     dict: escape_dict,
     bytearray: escape_bytes,
     type(None): escape_None,
diff --git a/pymysql/tests/test_basic.py b/pymysql/tests/test_basic.py
@@ -42,11 +42,15 @@ def test_datatypes(self):
 
             c.execute("delete from test_datatypes")
 
-            # check sequence type
-            c.execute("insert into test_datatypes (i, l) values (2,4), (6,8), (10,12)")
-            c.execute("select l from test_datatypes where i in %s order by i", ((2,6),))
-            r = c.fetchall()
-            self.assertEqual(((4,),(8,)), r)
+            # check sequences type
+            for seq_type in (tuple, list, set, frozenset):
+                c.execute("insert into test_datatypes (i, l) values (2,4), (6,8), (10,12)")
+                seq = seq_type([2,6])
+                c.execute("select l from test_datatypes where i in %s order by i", (seq,))
+                r = c.fetchall()
+                self.assertEqual(((4,),(8,)), r)
+                c.execute("delete from test_datatypes")
+
         finally:
             c.execute("drop table test_datatypes")
 
