code: def testMeanNonInfNaN(self):
      prob = tf.random.uniform([int(1e4)], seed=test_util.test_seed())
      dist = continuous_bernoulli.ContinuousBernoulli(
          probs=prob, validate_args=True
      )
      mean_ = self.evaluate(dist.mean())
      self.assertFalse(np.any(np.isinf(mean_)))
      self.assertFalse(np.any(np.isnan(mean_)))
  
imports: import numpy
  from scipy import special
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import continuous_bernoulli
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 21 for calling the ContinuousBernoulliTest obj's underlying member method code with example usage: obj.testMeanNonInfNaN() without return types