code: def test_kolmogorov_smirnov_distance_two_sample(self, dtype):
      del dtype
      samples1 = np.array([[1, 1, 2, 2], [1, 2, 3, 4], [1, 3, 3, 3]])
      samples2 = np.array(
          [[1, 2, 2, 1, 1, 2], [4, 4, 3, 2, 1, 1], [2, 2, 2, 1, -4, 7]]
      )
      # Unlike empirical_cdfs, the samples Tensors must come in iid across the
      # leading dimension.
      obtained = self.evaluate(
          st.kolmogorov_smirnov_distance_two_sample(
              tf.transpose(a=samples1), tf.transpose(a=samples2)
          )
      )
      self.assertAllClose(
          [0.0, 1 / 12.0, 0.75 - 1 / 6.0], obtained, atol=1e-7
      )
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 7 for calling the StatisticalTestingTest obj's underlying member method code with example usage: obj.test_kolmogorov_smirnov_distance_two_sample(dtype) without return types