|
19 | 19 | ReturnStatement, |
20 | 20 | TripleQuoted, |
21 | 21 | TrueStatement, |
22 | | - UnchangingLine, |
| 22 | + UnchangingLine, CstTypes, |
23 | 23 | ) |
24 | 24 |
|
25 | 25 | cstify_cst = ( |
26 | | - CommentStatement(line_no_start=1, line_no_end=1, value="# pragma: no cover"), |
27 | | - CommentStatement(line_no_start=1, line_no_end=2, value="\n# flake8: noqa"), |
28 | | - CommentStatement( |
29 | | - line_no_start=2, |
30 | | - line_no_end=3, |
31 | | - value="\n# === Copyright 2022 under CC0", |
32 | | - ), |
| 26 | + CommentStatement(line_no_start=1, line_no_end=2, value="# pragma: no cover\n"), |
| 27 | + CommentStatement(line_no_start=2, line_no_end=3, value="# flake8: noqa\n"), |
| 28 | + CommentStatement(line_no_start=3, line_no_end=4, value="# fmt: off\n"), |
| 29 | + CommentStatement(line_no_start=4, line_no_end=5, value="# === Copyright 2022 under CC0\n"), |
33 | 30 | TripleQuoted( |
34 | 31 | is_double_q=True, |
35 | 32 | is_docstr=False, |
36 | | - line_no_start=3, |
37 | | - line_no_end=4, |
38 | | - value='\n"""Module docstring goes here"""', |
39 | | - ), |
40 | | - FromStatement(line_no_start=4, line_no_end=6, value="\n\nfrom operator import add"), |
| 33 | + line_no_start=5, |
| 34 | + line_no_end=6, |
| 35 | + value='"""Module docstring goes here"""\n', |
| 36 | + ), |
| 37 | + UnchangingLine(line_no_start=6, line_no_end=7, value="\n"), |
| 38 | + FromStatement(line_no_start=7, line_no_end=8, value="from operator import add\n"), |
| 39 | + UnchangingLine(line_no_start=8, line_no_end=9, value="\n"), |
| 40 | + UnchangingLine(line_no_start=9, line_no_end=10, value="\n"), |
41 | 41 | ClassDefinitionStart( |
42 | | - line_no_start=6, line_no_end=9, value="\n\n\nclass C(object):", name="C" |
| 42 | + line_no_start=10, line_no_end=11, value="class C(object):\n", name="C" |
43 | 43 | ), |
44 | 44 | TripleQuoted( |
45 | 45 | is_double_q=True, |
46 | 46 | is_docstr=True, |
47 | | - line_no_start=9, |
48 | | - line_no_end=10, |
49 | | - value='\n """My cls"""', |
| 47 | + line_no_start=11, |
| 48 | + line_no_end=12, |
| 49 | + value=' """My cls"""\n', |
50 | 50 | ), |
| 51 | + UnchangingLine(line_no_start=12, line_no_end=13, value="\n"), |
51 | 52 | FunctionDefinitionStart( |
52 | | - line_no_start=10, |
53 | | - line_no_end=13, |
54 | | - value="\n" "\n" " @staticmethod\n" " def add1(foo):", |
| 53 | + line_no_start=13, |
| 54 | + line_no_end=15, |
| 55 | + value=" @staticmethod\n def add1(foo):\n", |
55 | 56 | name="add1", |
56 | 57 | ), |
57 | 58 | TripleQuoted( |
58 | 59 | is_double_q=True, |
59 | 60 | is_docstr=True, |
60 | | - line_no_start=13, |
61 | | - line_no_end=20, |
62 | | - value='\n """\n' |
| 61 | + line_no_start=15, |
| 62 | + line_no_end=22, |
| 63 | + value=' """\n' |
63 | 64 | " :param foo: a foo\n" |
64 | 65 | " :type foo: ```int```\n" |
65 | 66 | "\n" |
66 | 67 | " :return: foo + 1\n" |
67 | 68 | " :rtype: ```int```\n" |
68 | | - ' """', |
| 69 | + ' """\n', |
69 | 70 | ), |
| 71 | + UnchangingLine(line_no_start=22, line_no_end=23, value="\n"), |
70 | 72 | TripleQuoted( |
71 | 73 | is_double_q=True, |
72 | 74 | is_docstr=False, |
73 | | - line_no_start=20, |
74 | | - line_no_end=22, |
75 | | - value='\n\n """foo"""', |
| 75 | + line_no_start=23, |
| 76 | + line_no_end=24, |
| 77 | + value=' """foo"""\n', |
76 | 78 | ), |
77 | 79 | FunctionDefinitionStart( |
78 | | - line_no_start=22, line_no_end=23, value="\n def g():", name="g" |
79 | | - ), |
80 | | - TripleQuoted( |
81 | | - is_double_q=True, |
82 | | - is_docstr=True, |
83 | | - line_no_start=23, |
84 | | - line_no_end=23, |
85 | | - value=' """foo : bar ; can"""', |
| 80 | + line_no_start=24, line_no_end=25, value=' def g(): """foo : bar ; can"""; pass\n', |
| 81 | + name="g", |
86 | 82 | ), |
87 | | - PassStatement(line_no_start=23, line_no_end=23, value="; pass"), |
| 83 | + UnchangingLine(line_no_start=25, line_no_end=26, value="\n"), |
88 | 84 | FunctionDefinitionStart( |
89 | | - line_no_start=23, line_no_end=25, value="\n\n def h():", name="h" |
| 85 | + line_no_start=26, line_no_end=26, value=" def h(): # stuff\n", name="h" |
90 | 86 | ), |
91 | | - CommentStatement(line_no_start=25, line_no_end=25, value=" # stuff"), |
92 | | - PassStatement(line_no_start=25, line_no_end=26, value="\n pass"), |
| 87 | + PassStatement(line_no_start=27, line_no_end=28, value=" pass\n"), |
| 88 | + UnchangingLine(line_no_start=28, line_no_end=29, value="\n"), |
93 | 89 | FunctionDefinitionStart( |
94 | | - line_no_start=26, |
95 | | - line_no_end=29, |
96 | | - value="\n\n def adder(a: int,\n b: int) -> int:", |
| 90 | + line_no_start=29, |
| 91 | + line_no_end=31, |
| 92 | + value=" def adder(a: int,\n b: int) -> int:\n", |
97 | 93 | name="adder", |
98 | 94 | ), |
99 | 95 | TripleQuoted( |
100 | 96 | is_double_q=True, |
101 | 97 | is_docstr=True, |
102 | | - line_no_start=29, |
103 | | - line_no_end=36, |
104 | | - value="\n" |
105 | | - ' """\n' |
| 98 | + line_no_start=31, |
| 99 | + line_no_end=38, |
| 100 | + value=" \"\"\"\n" |
106 | 101 | " :param a: First arg\n" |
107 | 102 | "\n" |
108 | 103 | " :param b: Second arg\n" |
109 | 104 | "\n" |
110 | 105 | " :return: first + second arg\n" |
111 | | - ' """', |
112 | | - ), |
113 | | - CommentStatement( |
114 | | - line_no_start=36, line_no_end=37, value="\n # fmt: off" |
| 106 | + ' """\n', |
115 | 107 | ), |
| 108 | + CommentStatement(line_no_start=38, line_no_end=39, value=" # fmt: off\n"), |
116 | 109 | AnnAssignment( |
117 | | - line_no_start=37, |
118 | | - line_no_end=40, |
119 | | - value="\n res: \\\n int \\\n = a + b", |
| 110 | + line_no_start=39, |
| 111 | + line_no_end=42, |
| 112 | + value=" res: \\\n int \\\n = a + b\n", |
120 | 113 | ), |
121 | | - ReturnStatement(line_no_start=40, line_no_end=41, value="\n return res"), |
| 114 | + ReturnStatement(line_no_start=42, line_no_end=43, value=" return res\n"), |
| 115 | + UnchangingLine(line_no_start=43, line_no_end=44, value="\n"), |
122 | 116 | Assignment( |
123 | | - line_no_start=41, |
124 | | - line_no_end=47, |
125 | | - value="\n\n r = (\n add(foo, 1)\n or\n adder(foo, 1)\n )", |
126 | | - ), |
127 | | - IfStatement(line_no_start=47, line_no_end=48, value="\n if r:"), |
128 | | - NoneStatement(line_no_start=48, line_no_end=49, value="\n None"), |
129 | | - ElifStatement(line_no_start=49, line_no_end=50, value="\n elif r:"), |
130 | | - TrueStatement(line_no_start=50, line_no_end=51, value="\n True"), |
131 | | - FalseStatement(line_no_start=51, line_no_end=52, value="\n False"), |
| 117 | + line_no_start=44, |
| 118 | + line_no_end=49, |
| 119 | + value=" r = (\n" |
| 120 | + " add(foo, 1)\n" |
| 121 | + " or\n" |
| 122 | + " adder(foo, 1)\n" |
| 123 | + " )\n", |
| 124 | + ), |
| 125 | + IfStatement(line_no_start=49, line_no_end=50, value=" if r:\n"), |
| 126 | + NoneStatement(line_no_start=50, line_no_end=51, value=" None\n"), |
| 127 | + ElifStatement(line_no_start=51, line_no_end=52, value=" elif r:\n"), |
| 128 | + TrueStatement(line_no_start=52, line_no_end=53, value=" True\n"), |
| 129 | + FalseStatement(line_no_start=53, line_no_end=54, value=" False\n"), |
132 | 130 | CommentStatement( |
133 | | - line_no_start=52, line_no_end=53, value="\n # ([5,5] @ [5,5]) *\\" |
134 | | - ), |
135 | | - ExprStatement( |
136 | | - line_no_start=53, |
| 131 | + line_no_start=54, |
137 | 132 | line_no_end=55, |
138 | | - value="\n -5 / 7 ** 6 + \\\n 6.0 - 6e1 & 1+2.34j", |
139 | | - ), |
140 | | - AugAssignment(line_no_start=55, line_no_end=56, value="\n r <<= 5"), |
141 | | - CallStatement(line_no_start=56, line_no_end=57, value="\n print(r)"), |
142 | | - ElseStatement(line_no_start=57, line_no_end=58, value="\n else:"), |
143 | | - PassStatement(line_no_start=58, line_no_end=59, value="\n pass"), |
144 | | - CommentStatement(line_no_start=59, line_no_end=60, value="\n # fmt: on"), |
145 | | - CommentStatement( |
146 | | - line_no_start=60, |
147 | | - line_no_end=61, |
148 | | - value="\n # That^ incremented `foo` by 1", |
| 133 | + value=" # ([5,5] @ [5,5]) *\\\n", |
149 | 134 | ), |
150 | | - ReturnStatement(line_no_start=61, line_no_end=62, value="\n return r"), |
| 135 | + ExprStatement( |
| 136 | + line_no_start=55, |
| 137 | + line_no_end=57, |
| 138 | + value=" -5 / 7 ** 6 + \\\n 6.0 - 6e1 & 1+2.34j\n", |
| 139 | + ), |
| 140 | + AugAssignment(line_no_start=57, line_no_end=58, value=" r <<= 5\n"), |
| 141 | + CallStatement(line_no_start=58, line_no_end=59, value=" print(r)\n"), |
| 142 | + ElseStatement(line_no_start=59, line_no_end=60, value=" else:\n"), |
| 143 | + PassStatement(line_no_start=60, line_no_end=61, value=" pass\n"), |
| 144 | + CommentStatement(line_no_start=61, line_no_end=62, value=" # fmt: on\n"), |
151 | 145 | CommentStatement( |
152 | 146 | line_no_start=62, |
153 | | - line_no_end=65, |
154 | | - value="\n\n\n# from contextlib import ContextDecorator", |
| 147 | + line_no_end=63, |
| 148 | + value=" # That^ incremented `foo` by 1\n", |
155 | 149 | ), |
| 150 | + ReturnStatement(line_no_start=63, line_no_end=64, value=" return r\n"), |
| 151 | + UnchangingLine(line_no_start=64, line_no_end=65, value="\n"), |
| 152 | + UnchangingLine(line_no_start=65, line_no_end=66, value="\n"), |
156 | 153 | CommentStatement( |
157 | | - line_no_start=65, line_no_end=67, value="\n\n# with ContextDecorator():" |
| 154 | + line_no_start=66, |
| 155 | + line_no_end=67, |
| 156 | + value="# from contextlib import ContextDecorator\n", |
158 | 157 | ), |
159 | | - CommentStatement(line_no_start=67, line_no_end=68, value="\n# pass"), |
160 | | - FunctionDefinitionStart( |
161 | | - line_no_start=68, line_no_end=71, value="\n\n\ndef f():", name="f" |
| 158 | + UnchangingLine(line_no_start=67, line_no_end=68, value="\n"), |
| 159 | + CommentStatement( |
| 160 | + line_no_start=68, line_no_end=69, value="# with ContextDecorator():\n" |
162 | 161 | ), |
163 | | - ReturnStatement(line_no_start=71, line_no_end=72, value="\n return 1"), |
164 | | - UnchangingLine(line_no_start=72, line_no_end=73, value="\n"), |
| 162 | + CommentStatement(line_no_start=69, line_no_end=70, value="# pass\n"), |
| 163 | + UnchangingLine(line_no_start=70, line_no_end=71, value="\n"), |
| 164 | + UnchangingLine(line_no_start=71, line_no_end=72, value="\n"), |
| 165 | + FunctionDefinitionStart(line_no_start=72, line_no_end=73, value="def f():\n", name="f"), |
| 166 | + ReturnStatement(line_no_start=73, line_no_end=74, value=" return 1\n"), |
165 | 167 | ) # type: tuple[CstTypes, ...] |
166 | 168 |
|
167 | 169 |
|
|
0 commit comments