diff --git a/darwin/future/tests/meta/objects/test_stagemeta.py b/darwin/future/tests/meta/objects/test_stagemeta.py index 9a1d24c7d..65c0d28fc 100644 --- a/darwin/future/tests/meta/objects/test_stagemeta.py +++ b/darwin/future/tests/meta/objects/test_stagemeta.py @@ -81,3 +81,15 @@ def test_move_attached_files_to_stage( + f"v2/teams/default-team/items/ids?workflow_stage_ids={str(stage_meta.id)}&dataset_ids=1337", 1, ) + +def test_get_stage_id(stage_meta): + assert stage_meta.id == UUID("00000000-0000-0000-0000-000000000000") + +def test_get_stage_name(stage_meta): + assert stage_meta.name == "test-stage" + +def test_get_stage_type(stage_meta): + assert stage_meta.type == "annotate" + +def test_get_stage_edges(stage_meta): + assert stage_meta.edges == [] \ No newline at end of file