code: def test_pad():
      for device in ["cpu", "gpu"]:
          for batch_size in {1, 8}:
              for (
                  input_max_shape,
                  axes,
                  axis_names,
                  align,
                  shape_arg,
              ) in [
                  ((200, 400, 3), (0,), None, None, None),
                  ((200, 400, 3), None, "H", None, None),
                  ((200, 400, 3), (1,), None, None, None),
                  ((200, 400, 3), None, "W", None, None),
                  ((200, 400, 3), (0, 1), None, None, None),
                  ((200, 400, 3), None, "HW", None, None),
                  ((200, 400, 3), (), None, None, None),
                  ((200, 400, 3), [], None, None, None),
                  ((200, 400, 3), None, "", None, None),
                  ((200, 400, 3), (2,), None, (4,), None),
                  ((200, 400, 3), None, "C", (4,), None),
                  ((200, 400, 3), (0, 1), None, (256, 256), None),
                  ((200, 400, 3), None, "HW", (256, 256), None),
                  ((200, 400, 3), (0, 1), None, (16, 64), None),
                  ((200, 400, 3), None, "HW", (16, 64), None),
                  ((200, 400, 3), (0, 1), None, (256,), None),
                  ((200, 400, 3), None, "HW", (256,), None),
                  ((200, 400, 3), None, None, None, (-1, -1, 4)),
                  ((25, 100, 3), (0,), None, None, (25,)),
                  ((200, 400, 3), (0, 1), None, (4, 16), (1, 200)),
              ]:
                  yield (
                      check_pad,
                      device,
                      batch_size,
                      input_max_shape,
                      axes,
                      axis_names,
                      align,
                      shape_arg,
                  )
  
imports: import numpy
  import nvidia.dali
  import nvidia.dali.fn
  import nvidia.dali.math
  import nvidia.dali.ops
  import nvidia.dali.types
  from nvidia.dali import Pipeline, pipeline_def
  from nvidia.dali.backend_impl import TensorListGPU
  from nose_utils import assert_raises
  from test_utils import RandomlyShapedDataIterator, generator_random_axes_for_3d_input, generator_random_data, as_array
label_desc: Use image node_id 2 for calling a global function with example usage: test_pad() without return types