code: def test_single_binary_image(dtype):
      image = [
          [[1], [1], [1], [1], [1]],
          [[1], [1], [1], [1], [1]],
          [[0], [1], [0], [1], [0]],
          [[1], [0], [1], [0], [1]],
          [[0], [1], [0], [1], [0]],
      ]
      expected_output = np.array(
          [
              2,
              2.23606801,
              2,
              2.23606801,
              2,
              1,
              1.41421354,
              1,
              1.41421354,
              1,
              0,
              1,
              0,
              1,
              0,
              1,
              0,
              1,
              0,
              1,
              0,
              1,
              0,
              1,
              0,
          ]
      )
      image = tf.constant(image, dtype=tf.uint8)
  
      output = dist_ops.euclidean_dist_transform(image, dtype=dtype)
      output_flat = tf.reshape(output, [-1])
  
      assert output.dtype == dtype
      assert output.shape == [5, 5, 1]
      test_utils.assert_allclose_according_to_type(
          output_flat, 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 1 for calling a global function with example usage: test_single_binary_image(dtype) without return types