code: def test_sample_y_at_train_belongs(self, n_samples, atol):
      X, y = self.data_generator.random_dataset(n_samples)
  
      self.estimator.fit(X, y)
      y_ = self.estimator.sample_y(X)
      y_ = gs.reshape(gs.transpose(y_, [0, 2, 1]), (-1, y_.shape[1]))
  
      res = self.estimator.space.belongs(y_, atol=atol)
      expected = gs.ones(n_samples, dtype=bool)
      self.assertAllEqual(res, expected)
  
file_path: geomstats/geomstats/test_cases/learning/wrapped_gaussian_process.py
imports: import pytest
  import geomstats.backend
  from geomstats.test.random import RandomDataGenerator
  from geomstats.test_cases.learning._base import BaseEstimatorTestCase
label_desc: Use image node_id 5 for calling the WrappedGaussianProcessTestCase obj's underlying member method code with example usage: obj.test_sample_y_at_train_belongs(n_samples, atol) without return types