-
Notifications
You must be signed in to change notification settings - Fork 183
/
test_unnecessary_format_string.py
44 lines (38 loc) · 1.42 KB
/
test_unnecessary_format_string.py
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
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
#
from typing import Type
from libcst.codemod import Codemod, CodemodTest
from libcst.codemod.commands.unnecessary_format_string import UnnecessaryFormatString
class TestUnnecessaryFormatString(CodemodTest):
TRANSFORM: Type[Codemod] = UnnecessaryFormatString
def test_replace(self) -> None:
before = r"""
good: str = "good"
good: str = f"with_arg{arg}"
good = "good{arg1}".format(1234)
good = "good".format()
good = "good" % {}
good = "good" % ()
good = rf"good\d+{bar}"
good = f"wow i don't have args but don't mess my braces {{ up }}"
bad: str = f"bad" + "bad"
bad: str = f'bad'
bad: str = rf'bad\d+'
"""
after = r"""
good: str = "good"
good: str = f"with_arg{arg}"
good = "good{arg1}".format(1234)
good = "good".format()
good = "good" % {}
good = "good" % ()
good = rf"good\d+{bar}"
good = f"wow i don't have args but don't mess my braces {{ up }}"
bad: str = "bad" + "bad"
bad: str = 'bad'
bad: str = r'bad\d+'
"""
self.assertCodemod(before, after)