|
4 | 4 |
|
5 | 5 |
|
6 | 6 | def test_sort_onliner_tuple() -> None: |
7 | | - txt = dedent( |
8 | | - """\ |
| 7 | + txt = dedent("""\ |
9 | 8 | from mod import name1, name2 |
10 | 9 |
|
11 | 10 | __all__ = ("name2", "name1") |
12 | | - """ |
13 | | - ) |
| 11 | + """) |
14 | 12 |
|
15 | | - expected = dedent( |
16 | | - """\ |
| 13 | + expected = dedent("""\ |
17 | 14 | from mod import name1, name2 |
18 | 15 |
|
19 | 16 | __all__ = ("name1", "name2") |
20 | | - """ |
21 | | - ) |
| 17 | + """) |
22 | 18 |
|
23 | 19 | assert expected == sort_all._fix_src(txt, "<input>") |
24 | 20 |
|
25 | 21 |
|
26 | 22 | def test_sort_onliner_list() -> None: |
27 | | - txt = dedent( |
28 | | - """\ |
| 23 | + txt = dedent("""\ |
29 | 24 | from mod import name1, name2 |
30 | 25 |
|
31 | 26 | __all__ = ["name2", "name1"] |
32 | | - """ |
33 | | - ) |
| 27 | + """) |
34 | 28 |
|
35 | | - expected = dedent( |
36 | | - """\ |
| 29 | + expected = dedent("""\ |
37 | 30 | from mod import name1, name2 |
38 | 31 |
|
39 | 32 | __all__ = ["name1", "name2"] |
40 | | - """ |
41 | | - ) |
| 33 | + """) |
42 | 34 |
|
43 | 35 | assert expected == sort_all._fix_src(txt, "<input>") |
44 | 36 |
|
45 | 37 |
|
46 | 38 | def test_sort_onliner_set() -> None: |
47 | | - txt = dedent( |
48 | | - """\ |
| 39 | + txt = dedent("""\ |
49 | 40 | from mod import name1, name2 |
50 | 41 |
|
51 | 42 | __all__ = {"name2", "name1"} |
52 | | - """ |
53 | | - ) |
| 43 | + """) |
54 | 44 |
|
55 | | - expected = dedent( |
56 | | - """\ |
| 45 | + expected = dedent("""\ |
57 | 46 | from mod import name1, name2 |
58 | 47 |
|
59 | 48 | __all__ = {"name1", "name2"} |
60 | | - """ |
61 | | - ) |
| 49 | + """) |
62 | 50 |
|
63 | 51 | assert expected == sort_all._fix_src(txt, "<input>") |
64 | 52 |
|
65 | 53 |
|
66 | 54 | def test_sort_multiline_tuple() -> None: |
67 | | - txt = dedent( |
68 | | - """\ |
| 55 | + txt = dedent("""\ |
69 | 56 | from mod import name1, name2 |
70 | 57 |
|
71 | 58 | __all__ = ( |
72 | 59 | "name2", |
73 | 60 | "name1", |
74 | 61 | ) |
75 | | - """ |
76 | | - ) |
| 62 | + """) |
77 | 63 |
|
78 | | - expected = dedent( |
79 | | - """\ |
| 64 | + expected = dedent("""\ |
80 | 65 | from mod import name1, name2 |
81 | 66 |
|
82 | 67 | __all__ = ( |
83 | 68 | "name1", |
84 | 69 | "name2", |
85 | 70 | ) |
86 | | - """ |
87 | | - ) |
| 71 | + """) |
88 | 72 |
|
89 | 73 | assert expected == sort_all._fix_src(txt, "<input>") |
90 | 74 |
|
91 | 75 |
|
92 | 76 | def test_sort_multiline_list() -> None: |
93 | | - txt = dedent( |
94 | | - """\ |
| 77 | + txt = dedent("""\ |
95 | 78 | from mod import name1, name2 |
96 | 79 |
|
97 | 80 | __all__ = [ |
98 | 81 | "name2", |
99 | 82 | "name1", |
100 | 83 | ] |
101 | | - """ |
102 | | - ) |
| 84 | + """) |
103 | 85 |
|
104 | | - expected = dedent( |
105 | | - """\ |
| 86 | + expected = dedent("""\ |
106 | 87 | from mod import name1, name2 |
107 | 88 |
|
108 | 89 | __all__ = [ |
109 | 90 | "name1", |
110 | 91 | "name2", |
111 | 92 | ] |
112 | | - """ |
113 | | - ) |
| 93 | + """) |
114 | 94 |
|
115 | 95 | assert expected == sort_all._fix_src(txt, "<input>") |
116 | 96 |
|
117 | 97 |
|
118 | 98 | def test_sort_multiline_set() -> None: |
119 | | - txt = dedent( |
120 | | - """\ |
| 99 | + txt = dedent("""\ |
121 | 100 | from mod import name1, name2 |
122 | 101 |
|
123 | 102 | __all__ = { |
124 | 103 | "name2", |
125 | 104 | "name1", |
126 | 105 | } |
127 | | - """ |
128 | | - ) |
| 106 | + """) |
129 | 107 |
|
130 | | - expected = dedent( |
131 | | - """\ |
| 108 | + expected = dedent("""\ |
132 | 109 | from mod import name1, name2 |
133 | 110 |
|
134 | 111 | __all__ = { |
135 | 112 | "name1", |
136 | 113 | "name2", |
137 | 114 | } |
138 | | - """ |
139 | | - ) |
| 115 | + """) |
140 | 116 |
|
141 | 117 | assert expected == sort_all._fix_src(txt, "<input>") |
142 | 118 |
|
143 | 119 |
|
144 | 120 | def test_sort_skip_nonconst() -> None: |
145 | | - txt = dedent( |
146 | | - """\ |
| 121 | + txt = dedent("""\ |
147 | 122 | from mod import name1, name2 |
148 | 123 |
|
149 | 124 | __all__ = (name2, name1) |
150 | | - """ |
151 | | - ) |
| 125 | + """) |
152 | 126 |
|
153 | | - expected = dedent( |
154 | | - """\ |
| 127 | + expected = dedent("""\ |
155 | 128 | from mod import name1, name2 |
156 | 129 |
|
157 | 130 | __all__ = (name2, name1) |
158 | | - """ |
159 | | - ) |
| 131 | + """) |
160 | 132 |
|
161 | 133 | assert expected == sort_all._fix_src(txt, "<input>") |
162 | 134 |
|
163 | 135 |
|
164 | 136 | def test_sort_skip_non_list_set_tuple() -> None: |
165 | | - txt = dedent( |
166 | | - """\ |
| 137 | + txt = dedent("""\ |
167 | 138 | from mod import name1, name2 |
168 | 139 |
|
169 | 140 | __all__ = ['name1'] + ['name2'] |
170 | | - """ |
171 | | - ) |
| 141 | + """) |
172 | 142 |
|
173 | | - expected = dedent( |
174 | | - """\ |
| 143 | + expected = dedent("""\ |
175 | 144 | from mod import name1, name2 |
176 | 145 |
|
177 | 146 | __all__ = ['name1'] + ['name2'] |
178 | | - """ |
179 | | - ) |
| 147 | + """) |
180 | 148 |
|
181 | 149 | assert expected == sort_all._fix_src(txt, "<input>") |
182 | 150 |
|
183 | 151 |
|
184 | 152 | def test_sort_skip_non_unknown_type() -> None: |
185 | | - txt = dedent( |
186 | | - """\ |
| 153 | + txt = dedent("""\ |
187 | 154 | from mod import name1, name2 |
188 | 155 |
|
189 | 156 | __all__ = 123 |
190 | | - """ |
191 | | - ) |
| 157 | + """) |
192 | 158 |
|
193 | | - expected = dedent( |
194 | | - """\ |
| 159 | + expected = dedent("""\ |
195 | 160 | from mod import name1, name2 |
196 | 161 |
|
197 | 162 | __all__ = 123 |
198 | | - """ |
199 | | - ) |
| 163 | + """) |
200 | 164 |
|
201 | 165 | assert expected == sort_all._fix_src(txt, "<input>") |
202 | 166 |
|
203 | 167 |
|
204 | 168 | def test_skip_multiple_all() -> None: |
205 | | - txt = dedent( |
206 | | - """\ |
| 169 | + txt = dedent("""\ |
207 | 170 | from mod import name1, name2 |
208 | 171 |
|
209 | 172 | __all__ = ("name1") |
210 | 173 | __all__ = ("name2") |
211 | | - """ |
212 | | - ) |
| 174 | + """) |
213 | 175 |
|
214 | | - expected = dedent( |
215 | | - """\ |
| 176 | + expected = dedent("""\ |
216 | 177 | from mod import name1, name2 |
217 | 178 |
|
218 | 179 | __all__ = ("name1") |
219 | 180 | __all__ = ("name2") |
220 | | - """ |
221 | | - ) |
| 181 | + """) |
222 | 182 |
|
223 | 183 | assert expected == sort_all._fix_src(txt, "<input>") |
224 | 184 |
|
225 | 185 |
|
226 | 186 | def test_sort_ann_assign_simple() -> None: |
227 | | - txt = dedent( |
228 | | - """\ |
| 187 | + txt = dedent("""\ |
229 | 188 | from typing import Tuple |
230 | 189 | from mod import name1, name2 |
231 | 190 |
|
232 | 191 | __all__: Tuple[str, ...] |
233 | 192 | __all__ = ("name2", "name1") |
234 | | - """ |
235 | | - ) |
| 193 | + """) |
236 | 194 |
|
237 | | - expected = dedent( |
238 | | - """\ |
| 195 | + expected = dedent("""\ |
239 | 196 | from typing import Tuple |
240 | 197 | from mod import name1, name2 |
241 | 198 |
|
242 | 199 | __all__: Tuple[str, ...] |
243 | 200 | __all__ = ("name1", "name2") |
244 | | - """ |
245 | | - ) |
| 201 | + """) |
246 | 202 |
|
247 | 203 | assert expected == sort_all._fix_src(txt, "<input>") |
248 | 204 |
|
249 | 205 |
|
250 | 206 | def test_sort_ann_assign_real() -> None: |
251 | | - txt = dedent( |
252 | | - """\ |
| 207 | + txt = dedent("""\ |
253 | 208 | from typing import Tuple |
254 | 209 | from mod import name1, name2 |
255 | 210 |
|
256 | 211 | __all__: Tuple[str, ...] = ('name2', 'name1') |
257 | | - """ |
258 | | - ) |
| 212 | + """) |
259 | 213 |
|
260 | | - expected = dedent( |
261 | | - """\ |
| 214 | + expected = dedent("""\ |
262 | 215 | from typing import Tuple |
263 | 216 | from mod import name1, name2 |
264 | 217 |
|
265 | 218 | __all__: Tuple[str, ...] = ("name1", "name2") |
266 | | - """ |
267 | | - ) |
| 219 | + """) |
268 | 220 |
|
269 | 221 | assert expected == sort_all._fix_src(txt, "<input>") |
270 | 222 |
|
271 | 223 |
|
272 | 224 | def test_sort_aug_assign_real() -> None: |
273 | | - txt = dedent( |
274 | | - """\ |
| 225 | + txt = dedent("""\ |
275 | 226 | from mod import name1, name2 |
276 | 227 |
|
277 | 228 | __all__ = ('a', 'b') |
278 | 229 | __all__ += ('name2', 'name1') |
279 | | - """ |
280 | | - ) |
| 230 | + """) |
281 | 231 |
|
282 | | - expected = dedent( |
283 | | - """\ |
| 232 | + expected = dedent("""\ |
284 | 233 | from mod import name1, name2 |
285 | 234 |
|
286 | 235 | __all__ = ("a", "b") |
287 | 236 | __all__ += ("name1", "name2") |
288 | | - """ |
289 | | - ) |
| 237 | + """) |
290 | 238 |
|
291 | 239 | assert expected == sort_all._fix_src(txt, "<input>") |
292 | 240 |
|
293 | 241 |
|
294 | 242 | def test_sort_empty_tuple() -> None: |
295 | | - txt = dedent( |
296 | | - """\ |
| 243 | + txt = dedent("""\ |
297 | 244 | from mod import name1, name2 |
298 | 245 |
|
299 | 246 | __all__ = () |
300 | | - """ |
301 | | - ) |
| 247 | + """) |
302 | 248 |
|
303 | | - expected = dedent( |
304 | | - """\ |
| 249 | + expected = dedent("""\ |
305 | 250 | from mod import name1, name2 |
306 | 251 |
|
307 | 252 | __all__ = () |
308 | | - """ |
309 | | - ) |
| 253 | + """) |
310 | 254 |
|
311 | 255 | assert expected == sort_all._fix_src(txt, "<input>") |
312 | 256 |
|
313 | 257 |
|
314 | 258 | def test_sort_empty_list() -> None: |
315 | | - txt = dedent( |
316 | | - """\ |
| 259 | + txt = dedent("""\ |
317 | 260 | from mod import name1, name2 |
318 | 261 |
|
319 | 262 | __all__ = [] |
320 | | - """ |
321 | | - ) |
| 263 | + """) |
322 | 264 |
|
323 | | - expected = dedent( |
324 | | - """\ |
| 265 | + expected = dedent("""\ |
325 | 266 | from mod import name1, name2 |
326 | 267 |
|
327 | 268 | __all__ = [] |
328 | | - """ |
329 | | - ) |
| 269 | + """) |
330 | 270 |
|
331 | 271 | assert expected == sort_all._fix_src(txt, "<input>") |
0 commit comments