Add the Timer helper class in tools submodule

arkindex_worker/utils.py

+# -*- coding: utf-8 -*-
+import datetime
+from timeit import default_timer
+
+
+class Timer(object):
+    """
+    A context manager to help measure execution times
+    """
+
+    def __init__(self):
+        self.timer = default_timer
+
+    def __enter__(self):
+        self.start = self.timer()
+        return self
+
+    def __exit__(self, *args):
+        end = self.timer()
+        self.elapsed = end - self.start
+        self.delta = datetime.timedelta(seconds=self.elapsed)

tests/test_utils.py

+# -*- coding: utf-8 -*-
+from datetime import timedelta
+from time import sleep
+
+from arkindex_worker.utils import Timer
+
+
+def test_timer_type():
+    with Timer() as timer:
+        pass
+    assert isinstance(timer.delta, timedelta)
+
+
+def test_timer():
+    # Assert the second timer has recorded a longer period
+    with Timer() as timer:
+        pass
+    with Timer() as timer2:
+        sleep(1 / 100)
+    assert timer.delta < timer2.delta