Show More
@@ -5,7 +5,7 b'' | |||||
5 | # Distributed under the terms of the Modified BSD License. |
|
5 | # Distributed under the terms of the Modified BSD License. | |
6 |
|
6 | |||
7 |
|
7 | |||
8 | from collections import Counter, defaultdict, deque, OrderedDict |
|
8 | from collections import Counter, defaultdict, deque, OrderedDict, UserList | |
9 | import os |
|
9 | import os | |
10 | import pytest |
|
10 | import pytest | |
11 | import types |
|
11 | import types | |
@@ -294,6 +294,42 b' def test_basic_class():' | |||||
294 | assert type_pprint_wrapper.called is True |
|
294 | assert type_pprint_wrapper.called is True | |
295 |
|
295 | |||
296 |
|
296 | |||
|
297 | def test_collections_userlist(): | |||
|
298 | # Create userlist with cycle | |||
|
299 | a = UserList() | |||
|
300 | a.append(a) | |||
|
301 | ||||
|
302 | cases = [ | |||
|
303 | (UserList(), "UserList([])"), | |||
|
304 | ( | |||
|
305 | UserList(i for i in range(1000, 1020)), | |||
|
306 | "UserList([1000,\n" | |||
|
307 | " 1001,\n" | |||
|
308 | " 1002,\n" | |||
|
309 | " 1003,\n" | |||
|
310 | " 1004,\n" | |||
|
311 | " 1005,\n" | |||
|
312 | " 1006,\n" | |||
|
313 | " 1007,\n" | |||
|
314 | " 1008,\n" | |||
|
315 | " 1009,\n" | |||
|
316 | " 1010,\n" | |||
|
317 | " 1011,\n" | |||
|
318 | " 1012,\n" | |||
|
319 | " 1013,\n" | |||
|
320 | " 1014,\n" | |||
|
321 | " 1015,\n" | |||
|
322 | " 1016,\n" | |||
|
323 | " 1017,\n" | |||
|
324 | " 1018,\n" | |||
|
325 | " 1019])", | |||
|
326 | ), | |||
|
327 | (a, "UserList([UserList(...)])"), | |||
|
328 | ] | |||
|
329 | for obj, expected in cases: | |||
|
330 | assert pretty.pretty(obj) == expected | |||
|
331 | ||||
|
332 | ||||
297 | # TODO : pytest.mark.parametrise once nose is gone. |
|
333 | # TODO : pytest.mark.parametrise once nose is gone. | |
298 | def test_collections_defaultdict(): |
|
334 | def test_collections_defaultdict(): | |
299 | # Create defaultdicts with cycles |
|
335 | # Create defaultdicts with cycles |
General Comments 0
You need to be logged in to leave comments.
Login now