code: def testChiMean(self):
      df = np.arange(1, 6, dtype=np.float64)
      expected_mean = stats.chi.mean(df)
      dist = chi.Chi(df=df, validate_args=True)
      self.assertEqual(dist.mean().shape, df.shape)
      self.assertAllClose(self.evaluate(dist.mean()), expected_mean)
  
imports: import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import chi
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 5 for calling the ChiTest obj's underlying member method code with example usage: obj.testChiMean() without return types