mirror of
https://github.com/sphinx-doc/sphinx.git
synced 2025-02-25 18:55:22 -06:00
0b32e72635
Developers can write number literals in several ways. For example, decimal (1234), hexadecimal (0x1234), octal decimal (0o1234) and so on. But, AST module don't mind how the numbers written in the code. As a result, ast.unparse() could not reproduce the original form of number literals. This allows to construct number literals as possible using original source code. Note: This is only available in Python 3.8+.
78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
"""
|
|
test_pycode_ast
|
|
~~~~~~~~~~~~~~~
|
|
|
|
Test pycode.ast
|
|
|
|
:copyright: Copyright 2007-2016 by the Sphinx team, see AUTHORS.
|
|
:license: BSD, see LICENSE for details.
|
|
"""
|
|
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from sphinx.pycode import ast
|
|
|
|
|
|
@pytest.mark.parametrize('source,expected', [
|
|
("a + b", "a + b"), # Add
|
|
("a and b", "a and b"), # And
|
|
("os.path", "os.path"), # Attribute
|
|
("1 * 2", "1 * 2"), # BinOp
|
|
("a & b", "a & b"), # BitAnd
|
|
("a | b", "a | b"), # BitOr
|
|
("a ^ b", "a ^ b"), # BitXor
|
|
("a and b and c", "a and b and c"), # BoolOp
|
|
("b'bytes'", "b'bytes'"), # Bytes
|
|
("object()", "object()"), # Call
|
|
("1234", "1234"), # Constant
|
|
("{'key1': 'value1', 'key2': 'value2'}",
|
|
"{'key1': 'value1', 'key2': 'value2'}"), # Dict
|
|
("a / b", "a / b"), # Div
|
|
("...", "..."), # Ellipsis
|
|
("a // b", "a // b"), # FloorDiv
|
|
("Tuple[int, int]", "Tuple[int, int]"), # Index, Subscript
|
|
("~ 1", "~ 1"), # Invert
|
|
("lambda x, y: x + y",
|
|
"lambda x, y: ..."), # Lambda
|
|
("[1, 2, 3]", "[1, 2, 3]"), # List
|
|
("a << b", "a << b"), # LShift
|
|
("a @ b", "a @ b"), # MatMult
|
|
("a % b", "a % b"), # Mod
|
|
("a * b", "a * b"), # Mult
|
|
("sys", "sys"), # Name, NameConstant
|
|
("1234", "1234"), # Num
|
|
("not a", "not a"), # Not
|
|
("a or b", "a or b"), # Or
|
|
("a ** b", "a ** b"), # Pow
|
|
("a >> b", "a >> b"), # RShift
|
|
("{1, 2, 3}", "{1, 2, 3}"), # Set
|
|
("a - b", "a - b"), # Sub
|
|
("'str'", "'str'"), # Str
|
|
("+ a", "+ a"), # UAdd
|
|
("- 1", "- 1"), # UnaryOp
|
|
("- a", "- a"), # USub
|
|
("(1, 2, 3)", "(1, 2, 3)"), # Tuple
|
|
("()", "()"), # Tuple (empty)
|
|
])
|
|
def test_unparse(source, expected):
|
|
module = ast.parse(source)
|
|
assert ast.unparse(module.body[0].value, source) == expected
|
|
|
|
|
|
def test_unparse_None():
|
|
assert ast.unparse(None) is None
|
|
|
|
|
|
@pytest.mark.skipif(sys.version_info < (3, 8), reason='python 3.8+ is required.')
|
|
@pytest.mark.parametrize('source,expected', [
|
|
("lambda x=0, /, y=1, *args, z, **kwargs: x + y + z",
|
|
"lambda x=0, /, y=1, *args, z, **kwargs: ..."), # posonlyargs
|
|
("0x1234", "0x1234"), # Constant
|
|
("1_000_000", "1_000_000"), # Constant
|
|
])
|
|
def test_unparse_py38(source, expected):
|
|
module = ast.parse(source)
|
|
assert ast.unparse(module.body[0].value, source) == expected
|