import tensorflow as tf

    self.assertAllClose(masks.numpy(), expected_masks.numpy())

  def test_inputs_Distances_to_centers(self):
    inputs = tf.random.uniform(
        [100, 8], minval=-10, maxval=10.0, dtype=tf.float32)
    centers = tf.random.uniform(
