Encode UTF text only if in Python 2

pferate · pferate · commit ca397a7a27cc · 2015-03-09T12:34:49.000-07:00
diff --git a/tests/test_json_model.py b/tests/test_json_model.py
@@ -121,7 +121,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 six.PY3:
+      # 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, '{}')
