code: def test_auto_augmentation(
      test_image, augmentation_type, augmentation_params
  ):
      aug_config = get_augmentation_cls(
          IMAGE, augmentation_type
      ).from_dict(augmentation_params)
      augmentation_op_cls = get_augmentation_op(
          IMAGE, augmentation_type
      )
      augmentation_op = augmentation_op_cls(aug_config)
      augmented_image = augmentation_op(test_image)
      assert augmented_image.shape == (5, 3, 256, 256)
  
imports: import pytest
  import torch
  from ludwig.constants import IMAGE
  from ludwig.features.image_feature import get_augmentation_op
  from ludwig.schema.features.augmentation.utils import get_augmentation_cls
label_desc: Use image node_id 2 for calling a global function with example usage: test_auto_augmentation(test_image, augmentation_type, augmentation_params) without return types