def test_var_len_pos(self):
def add(a, b, *args):
return sum([a, b] + list(args))
add_ = curry(add)
assert add_(1, 2) == add_(a=1)(2) == add_(b=2)(1) == 3
assert add_(1, 2, 3) == add_(1)(2, 3) == 6
assert add_(1, 2, 3, 4) == 10
def test_var_len_kw(self):
def add(a, b, **kwargs):
return sum([a, b] + list(kwargs.values()))
add_ = curry(add)
assert add_(1, 2) == 3
assert add_(c=3)(1, 2) == 6 == add_(1, 2, c=3)
assert add_(1, 2, c=3, d=4) == 10
def test_simple(self):
def f(x, y, z):
return (x, y, z)
f_ = curry(f)
assert f_(1)(2)(3) == (1, 2, 3)
assert f_(x=1)(2)(3) == (1, 2, 3)
assert f_(x=1, y=2)(3) == (1, 2, 3)
assert f_(x=1, z=3)(2) == (1, 2, 3)
def test_builtin_func(self):
map_ = curry(map)
assert (list(map_(lambda x: x + 1)(range(10))) == list(
map_(lambda x: x + 1, range(10))) == list(range(1, 11)))
def is_odd(i):
return i % 2 != 0
filter_ = curry(filter)
assert (list(filter_(is_odd, range(10))) == list(
filter_(is_odd)(range(10))) == list(filter(is_odd, range(10))))
from functools import reduce
reduce_ = curry(reduce)
def add(a, b):
return a + b
assert reduce_(add, range(10)) == reduce_(add)(range(10)) == reduce(
add, range(10))
def test_with_default(self):
def f(x, y, z=1):
return x, y, z
f_ = curry(f)
assert f_(1)(2, 3) == (1, 2, 3)