modernize/tests/test_fix_import.py at master · PyCQA/modernize · GitHub

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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
from __future__ import generator_stop

from utils import check_on_input

NO_IMPORTS = (
    """\
""",
    """\
""",
)

ONLY_FUTURE_IMPORTS = (
    """\
from __future__ import print_function
""",
    """\
from __future__ import print_function
""",
)

ONLY_NORMAL_IMPORTS = (
    """\
import foo
""",
    """\
from __future__ import absolute_import
import foo
""",
)

NORMAL_AND_FUTURE_IMPORTS = (
    """\
from __future__ import print_function
import foo
""",
    """\
from __future__ import print_function
from __future__ import absolute_import
import foo
""",
)

DOCSTRING = (
    """\
\"""
Docstring
\"""
import foo
""",
    """\
\"""
Docstring
\"""
from __future__ import absolute_import
import foo
""",
)

SHEBANG = (
    """\
#!/usr/bin/env python
import foo
""",
    """\
#!/usr/bin/env python
from __future__ import absolute_import
import foo
""",
)

DOCSTING_AND_SHEBANG = (
    """\
#!/usr/bin/env python
\"""
Docstring
\"""
import foo
""",
    """\
#!/usr/bin/env python
\"""
Docstring
\"""
from __future__ import absolute_import
import foo
""",
)

COPYRIGHT_AND_SHEBANG = (
    """\
#!/usr/bin/env python

#
# Copyright notice
#

import foo
""",
    """\
#!/usr/bin/env python

#
# Copyright notice
#

from __future__ import absolute_import
import foo
""",
)


COPYRIGHT_AND_DOCSTRING = (
    """\
#
# Copyright notice
#

\"""Docstring\"""

import foo
""",
    """\
#
# Copyright notice
#

\"""Docstring\"""

from __future__ import absolute_import
import foo
""",
)


def test_no_imports():
    check_on_input(*NO_IMPORTS)


def test_only_future_imports():
    check_on_input(*ONLY_FUTURE_IMPORTS)


def test_only_normal_imports():
    check_on_input(*ONLY_NORMAL_IMPORTS)


def test_normal_and_future_imports():
    check_on_input(*NORMAL_AND_FUTURE_IMPORTS)


def test_import_with_docstring():
    check_on_input(*DOCSTRING)


def test_import_with_shebang():
    check_on_input(*SHEBANG)


def test_import_with_docstring_and_shebang():
    check_on_input(*DOCSTING_AND_SHEBANG)


def test_import_with_copyright_and_shebang():
    check_on_input(*COPYRIGHT_AND_SHEBANG)


def test_import_with_copyright_and_docstring():
    check_on_input(*COPYRIGHT_AND_DOCSTRING)