1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
from __future__ import annotations
import pytest
from pyupgrade._main import _fix_tokens
@pytest.mark.parametrize(
's',
(
'"☃".encode("UTF-8")',
'"\\u2603".encode("UTF-8")',
'"\\U0001f643".encode("UTF-8")',
'"\\N{SNOWMAN}".encode("UTF-8")',
'"\\xa0".encode("UTF-8")',
# not byte literal compatible
'"y".encode("utf16")',
# can't rewrite f-strings
'f"{x}".encode()',
# not a `.encode()` call
'"foo".encode', '("foo".encode)',
# encode, but not a literal
'x.encode()',
# the codec / string is an f-string
'str.encode(f"{c}")', '"foo".encode(f"{c}")',
pytest.param('wat.encode(b"unrelated")', id='unrelated .encode(...)'),
),
)
def test_binary_literals_noop(s):
assert _fix_tokens(s) == s
@pytest.mark.parametrize(
('s', 'expected'),
(
('"foo".encode()', 'b"foo"'),
('"foo".encode("ascii")', 'b"foo"'),
('"foo".encode("utf-8")', 'b"foo"'),
('"\\xa0".encode("latin1")', 'b"\\xa0"'),
(r'"\\u wot".encode()', r'b"\\u wot"'),
(r'"\\x files".encode()', r'b"\\x files"'),
(
'f(\n'
' "foo"\n'
' "bar".encode()\n'
')\n',
'f(\n'
' b"foo"\n'
' b"bar"\n'
')\n',
),
),
)
def test_binary_literals(s, expected):
assert _fix_tokens(s) == expected
|
