code: def test_expand_dims():
      # axes, new_axis_names, layout, shapes, expected_shapes, expected_layout
      args = [
          (
              [0, 2],
              "AB",
              "XYZ",
              [(10, 20, 30)],
              [(1, 10, 1, 20, 30)],
              "AXBYZ",
          ),
          (
              [0, 3],
              None,
              "XYZ",
              [(10, 20, 30)],
              [(1, 10, 20, 1, 30)],
              "",
          ),
          (
              [3],
              None,
              "XYZ",
              [(10, 20, 30), (100, 200, 300)],
              [(10, 20, 30, 1), (100, 200, 300, 1)],
              "",
          ),
          (
              [4, 3],
              None,
              "XYZ",
              [(10, 20, 30), (100, 200, 300)],
              [(10, 20, 30, 1, 1), (100, 200, 300, 1, 1)],
              "",
          ),
          (
              [0, 1, 3, 5, 7],
              "ABCDE",
              "XYZ",
              [(11, 22, 33)],
              [(1, 1, 11, 1, 22, 1, 33, 1)],
              "ABXCYDZE",
          ),
          ([], "", "HW", [(10, 20)], [(10, 20)], "HW"),
          ([0, 1], "", "", [()], [(1, 1)], ""),
          ([0], "", "HW", [(10, 20)], [(1, 10, 20)], ""),
          (
              [4, 3],
              "AB",
              "XYZ",
              [(10, 20, 30)],
              [(10, 20, 30, 1, 1)],
              "XYZBA",
          ),
          ([0], "X", "", [()], [(1,)], "X"),
      ]
      for (
          axes,
          new_axis_names,
          layout,
          shapes,
          expected_out_shapes,
          expected_layout,
      ) in args:
          yield (
              _testimpl_expand_dims,
              axes,
              new_axis_names,
              layout,
              shapes,
              expected_out_shapes,
              expected_layout,
          )
  
file_path: DALI/dali/test/python/operator_1/test_expand_dims.py
imports: from nvidia.dali import pipeline_def
  import nvidia.dali.fn
  import numpy
  from nose_utils import assert_raises
label_desc: Use image node_id 4 for calling a global function with example usage: test_expand_dims() without return types