Memoization fibonacci

memoization_fibonacci.py

import timeit
from functools import wraps


def memoize(func):
    cache = dict()
    @wraps(func)
    def to_cache(*args):
        if args in cache:
            return cache[args]
        result = func(*args)
        cache[args] = result
        return result
    return to_cache


@memoize
def fibonacci_memoize(n: int):
    if n == 0:
        return 0
    elif n == 1:
        return 1
    return fibonacci_memoize(n-1) + fibonacci_memoize(n-2)


N = 35
print(timeit.timeit("fibonacci_memoize(N)", globals=globals(), number=1))
# 5.855500000073732e-05