Timed

2020/day_8/handheld_halting.py

@@ -1,3 +1,5 @@
+import timeit
+
 def execute(memory):
     acc = 0 #accumulator
     ip = 0 #instruction pointer
@@ -42,6 +44,16 @@ def solve_b(memory):
 
     return "OH NO"
 
+def time(function):
+    start = timeit.default_timer()
+    result = function()
+    end = timeit.default_timer()
+    return ((end - start) * 1000, result)
+
 memory = [line.strip() for line in open('input')]
-print(solve_a(memory))
-print(solve_b(memory))
+
+elapsed, solution = time(lambda: solve_a(memory))
+print(f'Part N°1 [{elapsed:.4f}ms] = {solution}')
+
+elapsed, solution = time(lambda: solve_b(memory))
+print(f'Part N°2 [{elapsed:.4f}ms] = {solution}')