code: def test_random_transforms():
      x = np.random.random((2, 28, 28))
      assert affine_transformations.random_rotation(x, 45).shape == (
          2,
          28,
          28,
      )
      assert affine_transformations.random_shift(x, 1, 1).shape == (
          2,
          28,
          28,
      )
      assert affine_transformations.random_shear(x, 20).shape == (
          2,
          28,
          28,
      )
      assert affine_transformations.random_channel_shift(
          x, 20
      ).shape == (2, 28, 28)
  
file_path: keras-preprocessing/tests/image/affine_transformations_test.py
imports: import numpy
  import pytest
  from keras_preprocessing.image import affine_transformations
label_desc: Use image node_id 1 for calling a global function with example usage: test_random_transforms() without return types