docstr: Tests that function calculation correctly handles overflow.
docstr_tok: ["Tests","that","function","calculation","correctly","handles","overflow","."]
code: def test_log_soomean_exp_2(self):
      """Tests that function calculation correctly handles overflow."""
  
      # Using 700 (rather than 1e3) since naive numpy version can't handle higher.
      logx = np.float32([0.0, 700, -1, 1])
      np_log_soomean_u, np_log_mean_u = self._numpy_log_soomean_exp(
          logx, axis=0
      )
      [log_soomean_u, log_mean_u] = self.evaluate(
          leave_one_out.log_soomean_exp(logx, axis=0)
      )
      self.assertAllClose(
          np_log_mean_u, log_mean_u, rtol=1e-6, atol=0.0
      )
      self.assertAllClose(
          np_log_soomean_u, log_soomean_u, rtol=1e-5, atol=0.0
      )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.stats import leave_one_out
label_desc: Use image node_id 6 for calling the LogSooMeanTest obj's underlying member method code with example usage: obj.test_log_soomean_exp_2() without return types