From 464c116d95a613b17610098eecf2b49397cb0385 Mon Sep 17 00:00:00 2001 From: Knut Wannheden Date: Thu, 5 Sep 2024 21:11:40 +0200 Subject: [PATCH] More fixes for type hints --- rewrite/rewrite/python/_parser_visitor.py | 2 ++ rewrite/tests/python/all/literal_test.py | 5 +++++ 2 files changed, 7 insertions(+) diff --git a/rewrite/rewrite/python/_parser_visitor.py b/rewrite/rewrite/python/_parser_visitor.py index 707a165..2445583 100644 --- a/rewrite/rewrite/python/_parser_visitor.py +++ b/rewrite/rewrite/python/_parser_visitor.py @@ -1591,6 +1591,8 @@ def visit_UnaryOp(self, node): def __convert_type_hint(self, node) -> Optional[TypeTree]: if isinstance(node, ast.Constant): literal = cast(j.Literal, self.__convert(node)) + if node.value is None or node.value is Ellipsis: + return literal return j.Identifier( random_id(), literal.prefix, diff --git a/rewrite/tests/python/all/literal_test.py b/rewrite/tests/python/all/literal_test.py index 5f37b33..55fe3fe 100644 --- a/rewrite/tests/python/all/literal_test.py +++ b/rewrite/tests/python/all/literal_test.py @@ -35,6 +35,11 @@ def test_fraction_leading_dot(): rewrite_run(python("assert .0")) +def test_large_int(): + # language=python + rewrite_run(python("assert 0xC03A0019")) + + def test_single_quoted_string(): # language=python rewrite_run(python("assert 'foo'"))