code: def test_with_tf_function():
      test_image = tf.ones([1, 40, 40, 1], dtype=tf.uint8)
      result_image = tf.function(random_cutout)(test_image, 2)
      cutout_area = tf.zeros([4, 4], dtype=tf.uint8)
      cutout_area = tf.pad(
          cutout_area, ((0, 36), (0, 36)), constant_values=1
      )
      expect_image = to_4D_image(cutout_area)
      np.testing.assert_equal(result_image.shape, expect_image.shape)
  
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 4 for calling a global function with example usage: test_with_tf_function() without return types