mirror of
https://github.com/sphinx-doc/sphinx.git
synced 2025-02-25 18:55:22 -06:00
74 lines
2.9 KiB
Python
74 lines
2.9 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) == 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.')
|
|
def test_unparse_py38():
|
|
source = "lambda x=0, /, y=1, *args, z, **kwargs: x + y + z"
|
|
expected = "lambda x=0, /, y=1, *args, z, **kwargs: ..."
|
|
module = ast.parse(source)
|
|
assert ast.unparse(module.body[0].value) == expected
|