def test_to_none(self):
"""
Everything maps to None
"""
def to_none(a):
return None
init_list = range(10)
ok_list, bad_list = maps.partial_map(to_none, init_list)
assert ok_list == []
assert bad_list == range(10)
def test_all_ok(self):
"""
Everthing maps to a true value
"""
def id(a):
return a
init_list = range(1, 11)
ok_list, bad_list = maps.partial_map(id, init_list)
assert ok_list == init_list, (
"Expected: %s, got: %s" % (repr(init_list), repr(ok_list))
)
assert bad_list == []
def test_partial(self):
"""
Realistic case, part goes either way
"""
def kill_even(x):
if x % 2 == 0:
return None
return x*10
init_list = range(10, 20)
ok_list, bad_list = maps.partial_map(kill_even, init_list)
assert ok_list == [110, 130, 150, 170, 190]
assert bad_list == [10, 12, 14, 16, 18]
