Skip to content

Commit

Permalink
add tests
Browse files Browse the repository at this point in the history
Signed-off-by: Troy Chiu <[email protected]>
  • Loading branch information
troychiu committed Feb 24, 2025
1 parent 2421361 commit 14cb643
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions tests/flytekit/integration/remote/test_remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -664,6 +664,9 @@ def test_execute_workflow_with_maptask(register):
)
assert execution.outputs["o0"] == [4, 5, 6]
assert len(execution.node_executions["n0"].task_executions) == 1
assert len(execution.node_executions["n0"].task_executions[0].closure.metadata.external_resources) == len(d)
for i in range(len(d)):
assert execution.node_executions["n0"].task_executions[0].closure.metadata.external_resources[i].phase == 3 # SUCCEEDED

def test_execution_workflow_with_maptask_in_dynamic(register):
remote = FlyteRemote(Config.auto(config_file=CONFIG), PROJECT, DOMAIN)
Expand All @@ -680,6 +683,9 @@ def test_execution_workflow_with_maptask_in_dynamic(register):
assert execution.node_executions["n0"].subworkflow_node_executions is not None
assert "n0-0-dn0" in execution.node_executions["n0"].subworkflow_node_executions
assert len(execution.node_executions["n0"].subworkflow_node_executions["n0-0-dn0"].task_executions) == 1
assert len(execution.node_executions["n0"].subworkflow_node_executions["n0-0-dn0"].task_executions[0].closure.metadata.external_resources) == len(d)
for i in range(len(d)):
assert execution.node_executions["n0"].subworkflow_node_executions["n0-0-dn0"].task_executions[0].closure.metadata.external_resources[i].phase == 3 # SUCCEEDED


def test_executes_nested_workflow_dictating_interruptible(register):
Expand Down

0 comments on commit 14cb643

Please sign in to comment.