code: def test_all_ones():
      image = tf.ones([10, 10, 1], tf.uint8)
      output = dist_ops.euclidean_dist_transform(image)
      expected_output = np.full(
          [10, 10, 1], tf.math.sqrt(tf.float32.max)
      )
      np.testing.assert_allclose(output, expected_output)
  
file_path: addons/tensorflow_addons/image/tests/distance_transform_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.image import distance_transform
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 5 for calling a global function with example usage: test_all_ones() without return types