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]