code: def test_get_steps_in_sub_dag(
      pipeline_graph_get_sub_dag, step_name, expected_steps
  ):
      step = pipeline_graph_get_sub_dag.step_map.get(step_name)
      sub_steps = pipeline_graph_get_sub_dag.get_steps_in_sub_dag(step)
      assert sub_steps == expected_steps
  
imports: from __future__ import absolute_import
  import pytest
  from sagemaker.remote_function.job import _JobSettings
  from sagemaker.workflow.function_step import step
  from sagemaker.workflow.step_outputs import get_step
  from sagemaker.workflow.pipeline import Pipeline, PipelineGraph
  from sagemaker.workflow.condition_step import ConditionStep
  from sagemaker.workflow.conditions import ConditionEquals, ConditionIn, ConditionNot, ConditionOr, ConditionGreaterThan
  from sagemaker.workflow.execution_variables import ExecutionVariables
  from sagemaker.workflow.parameters import ParameterInteger, ParameterString
  from sagemaker.workflow._steps_compiler import CompiledStep
  from tests.unit.sagemaker.workflow.helpers import ordered, CustomStep, CustomStepCollection
label_desc: Use image node_id 20 for calling a global function with example usage: test_get_steps_in_sub_dag(pipeline_graph_get_sub_dag, step_name, expected_steps) without return types