FilesExpand file tree

 testing

/

test_pycode_ast.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

73 lines (62 loc) · 2.9 KB

 testing

/

test_pycode_ast.py

Copy path

Top

File metadata and controls

• Code
• Blame

73 lines (62 loc) · 2.9 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

"""

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