code: def test_mean_sigma_keepdims(axis):
      np.random.seed(0)
      a = np.random.random((4, 5, 6))
      mu1, sigma1 = mean_sigma(a, axis, keepdims=False)
      mu2, sigma2 = mean_sigma(a, axis, keepdims=True)
  
      assert_array_equal(mu1.ravel(), mu2.ravel())
      assert_array_equal(sigma1.ravel(), sigma2.ravel())
  
      assert_array_equal(np.broadcast(a, mu2).shape, a.shape)
      assert_array_equal(np.broadcast(a, sigma2).shape, a.shape)
  
file_path: astroML/astroML/stats/tests/test_stats.py
imports: import pytest
  import numpy
  from numpy.testing import assert_array_almost_equal, assert_array_equal, assert_allclose
  from astroML.stats import mean_sigma, median_sigmaG, sigmaG, fit_bivariate_normal
  from astroML.stats.random import bivariate_normal, trunc_exp, linear
label_desc: Use image node_id 2 for calling a global function with example usage: test_mean_sigma_keepdims(axis) without return types