Implement currying

Author: maxicot

currying/__init__.py

@@ -0,0 +1 @@
+

currying/src/__init__.py

@@ -0,0 +1 @@
+

currying/src/lib.py

@@ -0,0 +1,20 @@
+# Curry a given function.
+# If the definition uses *args, specify the amount given in a specific case
+def curry(f: callable, arity=None):
+    if arity is None:
+        arity = f.__code__.co_argcount
+    elif arity < 0:
+        raise Exception("negative arity")
+    elif arity < f.__code__.co_argcount:
+        raise Exception("specified arity is lesser than required")
+
+    if arity == 0:
+        return lambda: f()
+
+    def inner(*args):
+        if len(args) >= arity:
+            return f(*args)
+
+        return lambda arg: inner(*args, arg)
+
+    return lambda arg: inner(arg)

currying/tests/__init__.py

@@ -0,0 +1 @@
+

currying/tests/test.py

@@ -0,0 +1,34 @@
+from ..src.lib import curry
+import pytest
+
+
+def add_args(*args):
+    return sum(args)
+
+
+def add2(a, b):
+    return a + b
+
+
+def test_specified():
+    assert curry(add_args, 3)(1)(2)(3) == 6
+
+
+def test_negative():
+    with pytest.raises(Exception) as e:
+        curry(add2, -1)
+        assert e == "negative arity"
+
+
+def test_less():
+    with pytest.raises(Exception) as e:
+        curry(add2, 1)
+        assert e == "specified arity is lesser than required"
+
+
+def test_unspecified():
+    assert curry(add2)(1)(2) == 3
+
+
+def test_zero():
+    assert curry(add_args, 0)() == 0