add time format

Author: supercoderhawk

pysenal/utils/utils.py

@@ -131,3 +131,23 @@ def json_serialize(obj):
             return str(obj)
         except:
             raise TypeError(repr(obj) + ' is not JSON serializable')
+
+
+def format_time(seconds):
+    """
+    format seconds to time string
+    :param seconds: seconds, in float format
+    :return: formatted time string
+    """
+    h = int(seconds // 3600)
+    m = int(seconds % 3600 // 60)
+    s = seconds % 60
+
+    if h:
+        time_str = '{:d}h {:d}min {:.02f}s'.format(h, m, s)
+    elif m:
+        time_str = '{:d}min {:.02f}s'.format(m, s)
+    else:
+        time_str = '{:.02f}s'.format(s)
+
+    return time_str

tests/utils/test_utils.py

@@ -115,3 +115,12 @@ def test_json_serialize():
     assert json_serialize(Decimal('11.0')) == '11.0'
     assert json_serialize(b'123') == "b'123'"
     assert json_serialize(11) == '11'
+
+
+def test_format_time():
+    assert format_time(60) == '1min 0.00s'
+    assert format_time(60.1) == '1min 0.10s'
+    assert format_time(3600) == '1h 0min 0.00s'
+    assert format_time(3660.1121) == '1h 1min 0.11s'
+
+    assert format_time(12.2132145) == '12.21s'