Modernize code, add @cache example

Author: gjbex

source-code/decorators/memoize.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-from functools import lru_cache
+from functools import cache, lru_cache
 
 
 def memoize(f):
@@ -30,24 +30,32 @@ def fib_lru_cache(n):
         return fib_lru_cache(n - 1) + fib_lru_cache(n - 2)
 
 
+@cache
+def fib_cache(n):
+    if n < 2:
+        return 1
+    else:
+        return fib_cache(n - 1) + fib_cache(n - 2)
+
+
 def fib(n):
     if n < 2:
         return 1
     else:
         return fib(n - 1) + fib(n - 2)
 
 
-def execute(func, n_max):
+def execute(func, n_max, verbose=False):
     values = []
     start = datetime.now()
     for n in range(n_max):
         values.append(func(n))
     delta = datetime.now() - start
-    for n in range(n_max):
-        print('{0}({1}) = {2}'.format(func.__name__, n, values[n]))
-    delta_time = float(delta.seconds) + 1.0e-6*float(delta.microseconds)
-    print('{0}: {1:.6f} s'.format(func.__name__, delta_time))
-    return delta_time
+    if verbose:
+        for n in range(n_max):
+            print('{0}({1}) = {2}'.format(func.__name__, n, values[n]))
+    return float(delta.seconds) + 1.0e-6*float(delta.microseconds)
+
 
 if __name__ == '__main__':
     from argparse import ArgumentParser
@@ -56,10 +64,9 @@ def execute(func, n_max):
     arg_parser = ArgumentParser(description='compare memoized versus '
                                             'non-memooized')
     arg_parser.add_argument('n_max', type=int, help='maximum n value')
+    arg_parser.add_argument('--verbose', action='store_true',
+                            help='display computed values')
     options = arg_parser.parse_args()
-    delta_fib = execute(fib, options.n_max)
-    delta_fib_memoized = execute(fib_memoized, options.n_max)
-    delta_fib_lru_cache = execute(fib_lru_cache, options.n_max)
-    print('non-memoized:\t\t{0:.6f} s'.format(delta_fib))
-    print('memoized:\t\t{0:.6f} s'.format(delta_fib_memoized))
-    print('lru_cache:\t\t{0:.6f} s'.format(delta_fib_lru_cache))
+    for fib_impl in (fib, fib_memoized, fib_cache, fib_lru_cache):
+        delta_time = execute(fib_impl, options.n_max, options.verbose)
+        print(f'{fib_impl.__name__:20s}: {delta_time:.6f} s')