Skip to content

Commit

Permalink
update unit test
Browse files Browse the repository at this point in the history
Signed-off-by: Paul Dittamo <[email protected]>
  • Loading branch information
pvditt committed Nov 20, 2024
1 parent 35d5e93 commit 86bfe79
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tests/flytekit/unit/core/test_array_node.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from collections import OrderedDict

import pytest
from flyteidl.core import workflow_pb2 as _core_workflow

from flytekit import LaunchPlan, task, workflow
from flytekit.core.context_manager import FlyteContextManager
Expand Down Expand Up @@ -96,6 +97,10 @@ def test_lp_serialization(target, overrides_metadata, serialization_settings):

parent_node = wf_spec.template.nodes[0]
assert parent_node.inputs[0].var == "a"
assert parent_node.array_node._min_success_ratio == 0.9
assert parent_node.array_node._parallelism == 10
assert parent_node.array_node._is_original_sub_node_interface
assert parent_node.array_node._execution_mode == _core_workflow.ArrayNode.FULL_STATE
assert len(parent_node.inputs[0].binding.collection.bindings) == 3
for binding in parent_node.inputs[0].binding.collection.bindings:
assert binding.scalar.primitive.integer is not None
Expand Down

0 comments on commit 86bfe79

Please sign in to comment.