code: def testCholeskyFnType(self):
      identity = tf.linalg.LinearOperatorIdentity(3)
      self.assertIsInstance(
          cholesky_util.cholesky_from_fn(identity, tf.linalg.cholesky),
          tf.linalg.LinearOperatorIdentity,
      )
  
      diag = tf.linalg.LinearOperatorDiag([3.0, 5.0, 7.0])
      self.assertIsInstance(
          cholesky_util.cholesky_from_fn(diag, tf.linalg.cholesky),
          tf.linalg.LinearOperatorDiag,
      )
  
      kron = tf.linalg.LinearOperatorKronecker([identity, diag])
      self.assertIsInstance(
          cholesky_util.cholesky_from_fn(kron, tf.linalg.cholesky),
          tf.linalg.LinearOperatorKronecker,
      )
  
      block_diag = tf.linalg.LinearOperatorBlockDiag([identity, diag])
      self.assertIsInstance(
          cholesky_util.cholesky_from_fn(
              block_diag, tf.linalg.cholesky
          ),
          tf.linalg.LinearOperatorBlockDiag,
      )
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import cholesky_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the CholeskyUtilsTest obj's underlying member method code with example usage: obj.testCholeskyFnType() without return types