code: def test_matrix_center():
      x = np.expand_dims(
          np.array(
              [
                  [0, 1],
                  [0, 0],
              ]
          ),
          -1,
      )
      x_rotated90 = np.expand_dims(
          np.array(
              [
                  [1, 0],
                  [0, 0],
              ]
          ),
          -1,
      )
  
      assert np.allclose(
          affine_transformations.apply_affine_transform(
              x, theta=90, row_axis=0, col_axis=1, channel_axis=2
          ),
          x_rotated90,
      )
  
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 3 for calling a global function with example usage: test_matrix_center() without return types