Added unit tests

v7labs · Oct 18, 2023 · f08a319 · f08a319
1 parent 8b0e213
commit f08a319
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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 == []