Support byte string literal concatenation

Author: knutwannheden

rewrite/rewrite/python/_parser_visitor.py

@@ -1269,7 +1269,7 @@ def visit_Constant(self, node):
         while tok.type in (token.ENCODING, token.NL, token.NEWLINE, token.INDENT, token.DEDENT, token.COMMENT):
             tok = next(tokens)
 
-        if not isinstance(node.value, str):
+        if not isinstance(node.value, (str, bytes)):
             return self.__map_literal(node, tok, tokens)[0]
 
         is_byte_string = tok.string.startswith(('b', "B"))
@@ -1327,7 +1327,7 @@ def __map_literal(self, node, tok, tokens):
     def __map_literal_value(self, node, tok):
         if node.value is Ellipsis:
             return None
-        elif isinstance(node.value, str):
+        elif isinstance(node.value, (str, bytes)):
             return ast.literal_eval(ast.parse(tok.string, mode='eval').body)
         return node.value
 

rewrite/tests/python/all/literal_test.py

@@ -40,6 +40,11 @@ def test_large_int():
     rewrite_run(python("assert 0xC03A0019"))
 
 
+def test_byte_string_concatenation():
+    # language=python
+    rewrite_run(python("assert b'hello' b'world'"))
+
+
 def test_bigint():
     # language=python
     rewrite_run(python("assert 9223372036854775808"))