Skip to content

Commit f14ddf5

Browse files
jvstmeSergey Mezentsev
authored andcommitted
Add tests for reason and status conversions
1 parent 789f4b0 commit f14ddf5

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
from dstack._internal.core.models.runs import (
2+
JobStatus,
3+
JobTerminationReason,
4+
RunStatus,
5+
RunTerminationReason,
6+
)
7+
8+
9+
def test_run_to_job_termination_reason_works_with_all_enum_variants():
10+
for run_termination_reason in RunTerminationReason:
11+
job_termination_reason = run_termination_reason.to_job_termination_reason()
12+
assert isinstance(job_termination_reason, JobTerminationReason)
13+
14+
15+
def test_run_termination_reason_to_status_works_with_all_enum_variants():
16+
for run_termination_reason in RunTerminationReason:
17+
run_status = run_termination_reason.to_status()
18+
assert isinstance(run_status, RunStatus)
19+
20+
21+
def test_job_termination_reason_to_status_works_with_all_enum_varians():
22+
for job_termination_reason in JobTerminationReason:
23+
job_status = job_termination_reason.to_status()
24+
assert isinstance(job_status, JobStatus)

0 commit comments

Comments
 (0)