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
|
from __future__ import generator_stop
from utils import check_on_input
FILTER_CALL = (
"""\
filter(func, [1])
""",
"""\
from __future__ import absolute_import
from six.moves import filter
list(filter(func, [1]))
""",
)
FILTER_ITERATOR_CONTEXT = (
"""\
for a in filter(func, [1]):
pass
""",
"""\
from __future__ import absolute_import
from six.moves import filter
for a in filter(func, [1]):
pass
""",
)
FILTER_NONE = (
"""\
filter(None, x)
""",
"""\
[_f for _f in x if _f]
""",
)
def test_filter_call():
check_on_input(*FILTER_CALL)
def test_filter_iterator_context():
check_on_input(*FILTER_ITERATOR_CONTEXT)
def test_filter_None():
check_on_input(*FILTER_NONE)
|
