code: def test_estimate_belongs(self, n_samples, atol):
      X = self.data_generator.random_point(n_points=n_samples)
      means = self.estimator.fit(X).means_
      belongs = self.estimator.space.belongs(means, atol=atol)
      self.assertAllEqual(
          belongs, gs.ones(self.estimator.n_gaussians, dtype=bool)
      )
  
file_path: geomstats/geomstats/test_cases/learning/expectation_maximization.py
imports: import pytest
  import geomstats.backend
  from geomstats.test.test_case import TestCase
  from geomstats.test_cases.learning._base import BaseEstimatorTestCase
label_desc: Use image node_id 1 for calling the RiemannianEMTestCase obj's underlying member method code with example usage: obj.test_estimate_belongs(n_samples, atol) without return types