class_docstr: Test for data-loading utilities.
class_docstr_tok: ["Test","for","data-loading","utilities","."]
docstr: Test that leaves data are beta distribution parameters.
docstr_tok: ["Test","that","leaves","data","are","beta","distribution","parameters","."]
code: def test_leaves(self):
      """Test that leaves data are beta distribution parameters."""
      beta = BetaDistributions()
      beta_param, distrib_type = data_utils.load_leaves()
      result = beta.belongs(beta_param)
      self.assertTrue(gs.all(result))
  
      result = len(distrib_type)
      expected = beta_param.shape[0]
      self.assertAllClose(result, expected)
  
imports: import geomstats.backend
  import geomstats.datasets.utils
  from geomstats.geometry.euclidean import Euclidean
  from geomstats.geometry.hypersphere import Hypersphere
  from geomstats.geometry.landmarks import Landmarks
  from geomstats.geometry.spd_matrices import SPDMatrices
  from geomstats.geometry.special_euclidean import SpecialEuclidean
  from geomstats.geometry.special_orthogonal import SpecialOrthogonal
  from geomstats.information_geometry.beta import BetaDistributions
  from geomstats.test.test_case import TestCase, np_and_autograd_only
label_desc: Use image node_id 9 for calling the TestDatasets obj's underlying member method code with example usage: obj.test_leaves() without return types