2ef49d36a511bcc624720d0963b81226d61cacc1,tests/test_frechet_mean.py,TestFrechetMean,test_mean_and_belongs_hyperbolic,#TestFrechetMean#,93

Before Change

        point_c = self.space.random_uniform()
        points = gs.concatenate([point_a, point_b, point_c], axis=0)

        mean = self.metric.mean(points)
        result = self.space.belongs(mean)
        expected = gs.array([[True]])

        self.assertAllClose(result, expected)

After Change

        point_c = self.hyperbolic.random_uniform()
        points = gs.concatenate([point_a, point_b, point_c], axis=0)

        mean = FrechetMean(metric=self.hyperbolic.metric)
        mean.fit(X=points)

        result = self.hyperbolic.belongs(mean.mean_)
        expected = gs.array([[True]])

        self.assertAllClose(result, expected)

Instances