def METHOD_NAME(self):
    input_image = np.reshape(np.arange(0, 25), (5, 5, 1)).astype(np.float32)
    # 180 rotation.
    layer = RandomRotation(factor=(0.5, 0.5))
    output_image = layer(input_image)
    expected_output = np.asarray(
        [
            [24, 23, 22, 21, 20],
            [19, 18, 17, 16, 15],
            [14, 13, 12, 11, 10],
            [9, 8, 7, 6, 5],
            [4, 3, 2, 1, 0],
        ]
    ).astype(np.float32)
    expected_output = np.reshape(expected_output, (5, 5, 1))
    self.assertAllClose(expected_output, output_image)