Fixed a unicode encoding case

pferate · pferate · commit 58a7962bd214 · 2015-01-07T14:29:29.000-08:00
diff --git a/tests/test_json_model.py b/tests/test_json_model.py
@@ -123,7 +123,12 @@ def test_json_build_query(self):
 
     query_dict = parse_qs(query[1:])
     self.assertEqual(query_dict['foo'], ['1'])
-    self.assertEqual(query_dict['bar'], [u'\N{COMET}'.encode('utf-8')])
+    if isinstance(u'\N{COMET}', str):
+      # Python 3, no need to encode
+      self.assertEqual(query_dict['bar'], [u'\N{COMET}'])
+    else:
+      # Python 2, encode string
+      self.assertEqual(query_dict['bar'], [u'\N{COMET}'.encode('utf-8')])
     self.assertEqual(query_dict['baz'], ['fe', 'fi', 'fo', 'fum'])
     self.assertTrue('qux' not in query_dict)
     self.assertEqual(body, '{}')
