docstr: Check that the diagonal x1 = ... = xn is totally geodesic.
docstr_tok: ["Check","that","the","diagonal","x1","=","...","=","xn","is","totally","geodesic","."]
code: def test_exp_diagonal_is_totally_geodesic(
      self, param, tangent_param, atol
  ):
      """Check that the diagonal x1 = ... = xn is totally geodesic."""
      base_point = param * gs.ones(self.space.shape)
      tangent_vec = tangent_param * gs.ones(self.space.shape)
  
      end_point = self.space.metric.exp(tangent_vec, base_point)
      self.assertAllClose(
          end_point - end_point[0],
          gs.zeros(self.space.shape),
          atol=atol,
      )
  
file_path: geomstats/geomstats/test_cases/information_geometry/dirichlet.py
imports: import pytest
  import geomstats.backend
  from geomstats.test_cases.geometry.base import VectorSpaceOpenSetTestCase
  from geomstats.test_cases.geometry.riemannian_metric import RiemannianMetricTestCase
  from geomstats.test_cases.information_geometry.base import InformationManifoldMixinTestCase
label_desc: Use image node_id 3 for calling the DirichletMetricTestCase obj's underlying member method code with example usage: obj.test_exp_diagonal_is_totally_geodesic(param, tangent_param, atol) without return types