feat: add time limit parsing function and update time_to_run parameter to accept string inputs
This commit is contained in:
+13
-1
@@ -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")
|
||||
Reference in New Issue
Block a user