Encode UTF text only if in Python 2

googleapis · Mar 9, 2015 · ca397a7 · ca397a7
1 parent 2b14022
commit ca397a7
Showing 1 changed file with 6 additions and 1 deletion.
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, '{}')