Skip to content

Commit

Permalink
Add tests for reason and status conversions
Browse files Browse the repository at this point in the history
  • Loading branch information
jvstme authored and TheBits committed Mar 28, 2024
1 parent 789f4b0 commit f14ddf5
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions src/tests/_internal/core/models/test_runs.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
from dstack._internal.core.models.runs import (
JobStatus,
JobTerminationReason,
RunStatus,
RunTerminationReason,
)


def test_run_to_job_termination_reason_works_with_all_enum_variants():
for run_termination_reason in RunTerminationReason:
job_termination_reason = run_termination_reason.to_job_termination_reason()
assert isinstance(job_termination_reason, JobTerminationReason)


def test_run_termination_reason_to_status_works_with_all_enum_variants():
for run_termination_reason in RunTerminationReason:
run_status = run_termination_reason.to_status()
assert isinstance(run_status, RunStatus)


def test_job_termination_reason_to_status_works_with_all_enum_varians():
for job_termination_reason in JobTerminationReason:
job_status = job_termination_reason.to_status()
assert isinstance(job_status, JobStatus)

0 comments on commit f14ddf5

Please sign in to comment.