Function locate_operation

tests/flow/execution_plan_util.py:5–15

Searches for a specific operation in the execution plan and returns it if found, or None otherwise

(op: Operation, name: str)

def locate_operation(op: Operation, name: str):
    """Searches for a specific operation in the execution plan and returns it if
    found, or None otherwise"""
    if op.name == name:
        return op
    if op.children:
        for child in op.children:
            res = locate_operation(child, name)
            if res:
                return res
    return None

# Iterate recursively the operation's children and return the number of operations with a specific name
def count_operation(op: Operation, name: str):