Mark strings as unicode for Python 2

dimagi · Jun 10, 2020 · 529c4be · 529c4be
1 parent 284b8ee
commit 529c4be
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/tests/test_minilinq.py b/tests/test_minilinq.py
@@ -133,35 +133,35 @@ def test_template(self):
         assert Apply(Reference('template'), Literal('{}.{}'), Reference('a'), Reference('b')).eval(env) == '1.2'
 
     def test_substr(self):
-        env = BuiltInEnv({'single_byte_chars': 'abcdefghijklmnopqrstuvwxyz',
-                          'multi_byte_chars': 'αβγδεζηθικλμνξοπρςστυφχψω'
+        env = BuiltInEnv({'single_byte_chars': u'abcdefghijklmnopqrstuvwxyz',
+                          'multi_byte_chars': u'αβγδεζηθικλμνξοπρςστυφχψω'
         })
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
                      Literal(-4), Literal(30)).eval(env) == None
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
-                     Literal(0), Literal(26)).eval(env) == 'abcdefghijklmnopqrstuvwxyz'
+                     Literal(0), Literal(26)).eval(env) == u'abcdefghijklmnopqrstuvwxyz'
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
-                     Literal(10), Literal(16)).eval(env) == 'klmnop'
+                     Literal(10), Literal(16)).eval(env) == u'klmnop'
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
-                     Literal(13), Literal(14)).eval(env) == 'n'
+                     Literal(13), Literal(14)).eval(env) == u'n'
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
-                     Literal(13), Literal(13)).eval(env) == ''
+                     Literal(13), Literal(13)).eval(env) == u''
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
-                     Literal(14), Literal(13)).eval(env) == ''
+                     Literal(14), Literal(13)).eval(env) == u''
         assert Apply(Reference('substr'), Reference('single_byte_chars'),
                      Literal(5), Literal(-1)).eval(env) == None
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
                      Literal(-4), Literal(30)).eval(env) == None
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
-                     Literal(0), Literal(25)).eval(env) == 'αβγδεζηθικλμνξοπρςστυφχψω'
+                     Literal(0), Literal(25)).eval(env) == u'αβγδεζηθικλμνξοπρςστυφχψω'
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
-                     Literal(10), Literal(15)).eval(env) == 'λμνξο'
+                     Literal(10), Literal(15)).eval(env) == u'λμνξο'
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
-                     Literal(13), Literal(14)).eval(env) == 'ξ'
+                     Literal(13), Literal(14)).eval(env) == u'ξ'
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
-                     Literal(13), Literal(12)).eval(env) == ''
+                     Literal(13), Literal(12)).eval(env) == u''
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
-                     Literal(14), Literal(13)).eval(env) == ''
+                     Literal(14), Literal(13)).eval(env) == u''
         assert Apply(Reference('substr'), Reference('multi_byte_chars'),
                      Literal(5), Literal(-1)).eval(env) == None