changed time tracker representation to hh:mm:ss

src/helpers.py

@@ -13,6 +13,7 @@ import numpy as np
 import os
 import time
 import socket
+import datetime as dt
 
 
 def to_list(arg):
@@ -116,7 +117,8 @@ class TimeTracking(object):
             return time.time() - self.start
 
     def __repr__(self):
-        return f"{round(self._duration(), 2)}s"
+        # return f"{round(self._duration(), 2)}s"
+        return f"{dt.timedelta(seconds=math.ceil(self._duration()))} (hh:mm:ss)"
 
     def run(self):
         self._start()

test/test_helpers.py

@@ -117,7 +117,7 @@ class TestTimeTracking:
         t = TimeTracking()
         t._end()
         duration = t._duration()
-        assert t.__repr__().rstrip() == f"{round(duration, 2)}s".rstrip()
+        assert t.__repr__().rstrip() == f"{dt.timedelta(seconds=math.ceil(duration))} (hh:mm:ss)".rstrip()
 
     def test_run(self):
         t = TimeTracking(start=False)

test/test_modules/test_run_environment.py

@@ -16,7 +16,8 @@ class TestRunEnvironment:
         caplog.set_level(logging.INFO)
         with RunEnvironment() as r:
             r.do_stuff(0.1)
-        assert caplog.record_tuples[-1] == ('root', 20, PyTestRegex(r"RunEnvironment finished after \d+\.\d+s"))
+        expression = PyTestRegex(r"RunEnvironment finished after \d+:\d+:\d+ \(hh:mm:ss\)")
+        assert caplog.record_tuples[-1] == ('root', 20, expression)
 
     def test_init(self, caplog):
         caplog.set_level(logging.INFO)
@@ -28,4 +29,5 @@ class TestRunEnvironment:
         r = RunEnvironment()
         r.do_stuff(0.2)
         del r
-        assert caplog.record_tuples[-1] == ('root', 20, PyTestRegex(r"RunEnvironment finished after \d+\.\d+s"))
+        expression = PyTestRegex(r"RunEnvironment finished after \d+:\d+:\d+ \(hh:mm:ss\)")
+        assert caplog.record_tuples[-1] == ('root', 20, expression)