From 11a45212911b209d44fce8b2137265a5b259ab1e Mon Sep 17 00:00:00 2001 From: Frank Chen Date: Mon, 6 Jul 2026 12:08:35 -0700 Subject: [PATCH] test: allow parallel branch observation ordering --- .../test_parallel_with_named_branches.py | 21 ++++++++----------- 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/packages/aws-durable-execution-sdk-python-examples/test/parallel/test_parallel_with_named_branches.py b/packages/aws-durable-execution-sdk-python-examples/test/parallel/test_parallel_with_named_branches.py index 6b801b87..9fcc4382 100644 --- a/packages/aws-durable-execution-sdk-python-examples/test/parallel/test_parallel_with_named_branches.py +++ b/packages/aws-durable-execution-sdk-python-examples/test/parallel/test_parallel_with_named_branches.py @@ -38,18 +38,15 @@ def test_parallel_with_named_branches(durable_runner): # Verify branch names: named branches have custom names, unnamed use defaults assert len(parallel_op.child_operations) == 5 - child_names = [op.name for op in parallel_op.child_operations] - - # 1. Named ParallelBranch - assert child_names[0] == "fetch-user-data" - # 2. Named decorator - assert child_names[1] == "fetch-orders" - # 3. Unnamed decorator (None name falls back to index-based default) - assert child_names[2] == "parallel-branch-2" - # 4. Unnamed ParallelBranch (None name falls back to index-based default) - assert child_names[3] == "parallel-branch-3" - # 5. Raw callable (no ParallelBranch wrapper, index-based default) - assert child_names[4] == "parallel-branch-4" + child_names = {op.name for op in parallel_op.child_operations} + + assert child_names == { + "fetch-user-data", # Named ParallelBranch + "fetch-orders", # Named decorator + "parallel-branch-2", # Unnamed decorator + "parallel-branch-3", # Unnamed ParallelBranch + "parallel-branch-4", # Raw callable + } # Verify all children succeeded for child in parallel_op.child_operations: