code: def testLogProbWithZeroOrOneProbs(self):
      probs = [1.0, 0.0]  # logits = [np.inf, -np.inf]
      dist = bernoulli.Bernoulli(probs=probs)
      self.assertAllEqual([0.0, -np.inf], dist.log_prob([1.0, 1.0]))
      self.assertAllEqual([-np.inf, 0.0], dist.log_prob([0.0, 0.0]))
      self.assertAllEqual(
          [np.nan, np.nan], dist.log_prob([np.nan, np.nan])
      )
  
imports: import numpy
  from scipy import special
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import bernoulli
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.internal import samplers
  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 28 for calling the BernoulliTest obj's underlying member method code with example usage: obj.testLogProbWithZeroOrOneProbs() without return types