actions/black
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use is_number_token instead of assertion (#3069)

Browse Source
This commit is contained in:
Marco Edward Gorelli 2022-06-11 17:04:09 +01:00 committed by GitHub
parent 8c8675c62a
commit 162ecd1d2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/black/__init__.py
View File

@ -40,7 +40,7 @@
from black.const import DEFAULT_LINE_LENGTH, DEFAULT_INCLUDES, DEFAULT_EXCLUDES from black.const import DEFAULT_LINE_LENGTH, DEFAULT_INCLUDES, DEFAULT_EXCLUDES
from black.const import STDIN_PLACEHOLDER from black.const import STDIN_PLACEHOLDER
from black.nodes import STARS, syms, is_simple_decorator_expression from black.nodes import STARS, syms, is_simple_decorator_expression
from black.nodes import is_string_token from black.nodes import is_string_token, is_number_token
from black.lines import Line, EmptyLineTracker from black.lines import Line, EmptyLineTracker
from black.linegen import transform_line, LineGenerator, LN from black.linegen import transform_line, LineGenerator, LN
from black.comments import normalize_fmt_off from black.comments import normalize_fmt_off
@ -1245,8 +1245,7 @@ def get_features_used( # noqa: C901
if value_head in {'f"', 'F"', "f'", "F'", "rf", "fr", "RF", "FR"}: if value_head in {'f"', 'F"', "f'", "F'", "rf", "fr", "RF", "FR"}:
features.add(Feature.F_STRINGS) features.add(Feature.F_STRINGS)
elif n.type == token.NUMBER: elif is_number_token(n):
assert isinstance(n, Leaf)
if "_" in n.value: if "_" in n.value:
features.add(Feature.NUMERIC_UNDERSCORES) features.add(Feature.NUMERIC_UNDERSCORES)

4
src/black/nodes.py
View File

@ -854,3 +854,7 @@ def is_rpar_token(nl: NL) -> TypeGuard[Leaf]:
def is_string_token(nl: NL) -> TypeGuard[Leaf]: def is_string_token(nl: NL) -> TypeGuard[Leaf]:
return nl.type == token.STRING return nl.type == token.STRING
def is_number_token(nl: NL) -> TypeGuard[Leaf]:
return nl.type == token.NUMBER