feat: add time limit parsing function and update time_to_run parameter to accept string inputs

This commit is contained in:
Ross
2026-07-14 22:00:26 +01:00
parent 0874f05126
commit d577b2cdf8
3 changed files with 65 additions and 8 deletions
+13 -1
View File
@@ -2118,4 +2118,16 @@ def test_shift_dependent_dates_model_definition():
assert len(worker.shift_end_dates) == 1
assert isinstance(worker.shift_end_dates[0], ShiftEndDate)
assert worker.shift_end_dates[0].shift == "a"
assert worker.shift_end_dates[0].end_date == datetime.date(2022, 4, 7)
assert worker.shift_end_dates[0].end_date == datetime.date(2022, 4, 7)
def test_parse_time_limit():
from gen_proc import parse_time_limit
assert parse_time_limit(3600) == 3600
assert parse_time_limit("3600") == 3600
assert parse_time_limit("10h") == 36000
assert parse_time_limit("30m") == 1800
assert parse_time_limit("45s") == 45
assert parse_time_limit(" 1.5h ") == 5400
with pytest.raises(ValueError):
parse_time_limit("invalid")