Skip to content

Commit ed7b183

Browse files
committed
fix unhugging of tstring
1 parent 47f9f98 commit ed7b183

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

add_trailing_comma/_token_helpers.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,10 @@ def fix_brace(
136136
tokens[first_brace + 1].name == 'FSTRING_START' and
137137
tokens[last_brace - 1].name == 'FSTRING_END'
138138
) or
139+
(
140+
tokens[first_brace + 1].name == 'TSTRING_START' and
141+
tokens[last_brace - 1].name == 'TSTRING_END'
142+
) or
139143
# don't unhug if it is a single line
140144
fix_data.remove_comma
141145
):

tests/features/unhug_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,10 @@
2727
'textwrap.dedent(f"""\n'
2828
' hi\n'
2929
'""")',
30+
# single triple-quoted tstring argument, don't unhug
31+
'textwrap.dedent(t"""\n'
32+
' hi\n'
33+
'""")',
3034
),
3135
)
3236
def test_noop_unhugs(src):

0 commit comments

Comments
 (0)