code: def test_chunking(self):
      rng = test_util.test_np_rng()
      x = rng.rand(100, 10, 5)
      running_mean = sample_stats.RunningMean.from_shape(shape=(5,))
      running_mean = consume(running_mean, x, chunk_axis=0)
      mean = self.evaluate(running_mean.mean)
      self.assertAllClose(
          np.mean(x.reshape(1000, 5), axis=0), mean, rtol=1e-6
      )
  
imports: from absl.testing import parameterized
  import numpy
  import scipy.stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.stats import sample_stats
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import diagnostic
label_desc: Use image node_id 5 for calling the RunningMeanTest obj's underlying member method code with example usage: obj.test_chunking() without return types