3ef34a25874548e97d9a6e0aeeb2f4e71fb6c1fa,tests/test_special_euclidean_group.py,TestSpecialEuclideanGroupMethods,test_exp_vectorization,#TestSpecialEuclideanGroupMethods#,300

Before Change

                                        (n_samples, self.group.dimension)))

            // Test with the same number of base_points and tangent_vecs
            tangent_vecs = self.group.random_uniform(n_samples=n_samples)
            base_points = self.group.random_uniform(n_samples=n_samples)
            results = metric.exp(tangent_vecs, base_points)

            self.assertTrue(np.allclose(results.shape,
                                        (n_samples, self.group.dimension)))

After Change

        // TODO(nina): this test fails
        n_samples = self.n_samples
        for metric_type in self.metrics:
            metric = self.metrics[metric_type]

            one_tangent_vec = self.group.random_uniform(n_samples=1)
            one_base_point = self.group.random_uniform(n_samples=1)
            n_tangent_vec = self.group.random_uniform(n_samples=n_samples)
            n_base_point = self.group.random_uniform(n_samples=n_samples)

            // Test with the 1 base point, and n tangent vecs
            result = metric.exp(n_tangent_vec, one_base_point)
            self.assertTrue(np.allclose(result.shape,
                                        (n_samples, self.group.dimension)))
            expected = np.vstack([metric.exp(tangent_vec, one_base_point)
                                  for tangent_vec in n_tangent_vec])

Instances