code: def test_batch_size():
      test_image = tf.random.uniform(
          [10, 40, 40, 1], dtype=np.float32, seed=1234
      )
      result_image = random_cutout(test_image, 20, seed=1234)
      np.testing.assert_allclose(
          tf.shape(result_image), [10, 40, 40, 1]
      )
      means = np.mean(result_image, axis=(1, 2, 3))
      np.testing.assert_allclose(len(set(means)), 10)
  
file_path: addons/tensorflow_addons/image/tests/cutout_ops_test.py
imports: import pytest
  import tensorflow
  import numpy
  from tensorflow_addons.image.cutout_ops import cutout, random_cutout
  from tensorflow_addons.image.utils import to_4D_image
label_desc: Use image node_id 3 for calling a global function with example usage: test_batch_size() without return types