code: def testRegistration(self):
      class MyDist(normal.Normal):
          pass
  
      # Register KL to a lambda that spits out the name parameter
      @kullback_leibler.RegisterKL(MyDist, MyDist)
      def _kl(
          a, b, name=None
      ):  # pylint: disable=unused-argument,unused-variable
          return name
  
      a = MyDist(loc=0.0, scale=1.0)
      self.assertEqual(
          "OK", kullback_leibler.kl_divergence(a, a, name="OK")
      )
  
file_path: tensorflow/tensorflow/python/kernel_tests/distributions/kullback_leibler_test.py
imports: from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops.distributions import kullback_leibler
  from tensorflow.python.ops.distributions import normal
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the KLTest obj's underlying member method code with example usage: obj.testRegistration() and returns: name