code: def test_correct_distance():
      """Compare against numpy caluclation."""
      tf_embeddings = tf.constant([[0.5, 0.5], [1.0, 1.0]])
  
      expected_distance = np.array(
          [[0, np.sqrt(2) / 2], [np.sqrt(2) / 2, 0]]
      )
  
      distances = pairwise_distance(tf_embeddings, squared=False)
      np.testing.assert_allclose(
          expected_distance, distances, 1e-6, 1e-6
      )
  
file_path: addons/tensorflow_addons/losses/tests/metric_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.losses.metric_learning import pairwise_distance
label_desc: Use image node_id 3 for calling a global function with example usage: test_correct_distance() without return types