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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
import hypothesis
from hypothesis import strategies as st
import isort
CODE_SNIPPET = """
'''Taken from bottle.py
Copyright (c) 2009-2018, Marcel Hellkamp.
License: MIT (see LICENSE for details)
'''
# Lots of stdlib and builtin differences.
if py3k:
import http.client as httplib
import _thread as thread
from urllib.parse import urljoin, SplitResult as UrlSplitResult
from urllib.parse import urlencode, quote as urlquote, unquote as urlunquote
urlunquote = functools.partial(urlunquote, encoding='latin1')
from http.cookies import SimpleCookie, Morsel, CookieError
from collections.abc import MutableMapping as DictMixin
import pickle # comment number 2
from io import BytesIO
import configparser
basestring = str
unicode = str
json_loads = lambda s: json_lds(touni(s))
callable = lambda x: hasattr(x, '__call__')
imap = map
def _raise(*a):
raise a[0](a[1]).with_traceback(a[2])
else: # 2.x
import httplib
import thread
from urlparse import urljoin, SplitResult as UrlSplitResult
from urllib import urlencode, quote as urlquote, unquote as urlunquote
from Cookie import SimpleCookie, Morsel, CookieError
from itertools import imap
import cPickle as pickle
from StringIO import StringIO as BytesIO
import ConfigParser as configparser # commentnumberone
from collections import MutableMapping as DictMixin
unicode = unicode
json_loads = json_lds
exec(compile('def _raise(*a): raise a[0], a[1], a[2]', '<py3fix>', 'exec'))
"""
SHOULD_RETAIN = [
"""'''Taken from bottle.py
Copyright (c) 2009-2018, Marcel Hellkamp.
License: MIT (see LICENSE for details)
'''""",
"# Lots of stdlib and builtin differences.",
"if py3k:",
"http.client",
"_thread",
"urllib.parse",
"urlencode",
"urlunquote = functools.partial(urlunquote, encoding='latin1')",
"http.cookies",
"SimpleCookie",
"collections.abc",
"pickle",
"comment number 2",
"io",
"configparser",
"""basestring = str
unicode = str
json_loads = lambda s: json_lds(touni(s))
callable = lambda x: hasattr(x, '__call__')
imap = map
def _raise(*a):
raise a[0](a[1]).with_traceback(a[2])
else: # 2.x
""",
"httplib",
"thread",
"urlparse",
"urllib",
"Cookie",
"itertools",
"cPickle",
"StringIO",
"ConfigParser",
"commentnumberone",
"collections",
"""unicode = unicode
json_loads = json_lds
exec(compile('def _raise(*a): raise a[0], a[1], a[2]', '<py3fix>', 'exec'))""",
]
@hypothesis.given(
config=st.from_type(isort.Config), disregard_skip=st.booleans(),
)
def test_isort_is_idempotent(config: isort.Config, disregard_skip: bool) -> None:
try:
result = isort.code(CODE_SNIPPET, config=config, disregard_skip=disregard_skip)
result = isort.code(result, config=config, disregard_skip=disregard_skip)
assert result == isort.code(result, config=config, disregard_skip=disregard_skip)
except ValueError:
pass
@hypothesis.given(
config=st.from_type(isort.Config), disregard_skip=st.booleans(),
)
def test_isort_doesnt_lose_imports_or_comments(config: isort.Config, disregard_skip: bool) -> None:
result = isort.code(CODE_SNIPPET, config=config, disregard_skip=disregard_skip)
for should_be_retained in SHOULD_RETAIN:
if should_be_retained not in result:
if config.ignore_comments and should_be_retained.startswith("comment"):
continue
assert should_be_retained in result
|
