class TestVisualization(geomstats.tests.TestCase):
def setup_method(self):
self.n_samples = 10
self.SO3_GROUP = SpecialOrthogonal(n=3, point_type="vector")
self.SE3_GROUP = SpecialEuclidean(n=3, point_type="vector")
self.S1 = Hypersphere(dim=1)
self.S2 = Hypersphere(dim=2)
self.H2 = Hyperbolic(dim=2)
self.H2_half_plane = PoincareHalfSpace(dim=2)
self.M32 = Matrices(m=3, n=2)
self.S32 = PreShapeSpace(k_landmarks=3, m_ambient=2)
self.KS = visualization.KendallSphere()
self.M33 = Matrices(m=3, n=3)
self.S33 = PreShapeSpace(k_landmarks=3, m_ambient=3)
self.KD = visualization.KendallDisk()
self.spd = SPDMatrices(n=2)
plt.figure()
@staticmethod
def test_tutorial_matplotlib():
visualization.tutorial_matplotlib()
def test_plot_points_so3(self):
points = self.SO3_GROUP.random_uniform(self.n_samples)
visualization.plot(points, space="SO3_GROUP")
def test_plot_points_se3(self):
points = self.SE3_GROUP.random_point(self.n_samples)
visualization.plot(points, space="SE3_GROUP")
def test_draw_pre_shape_2d(self):
self.KS.draw()
def test_draw_points_pre_shape_2d(self):
points = self.S32.random_point(self.n_samples)
visualization.plot(points, space="S32")
points = self.M32.random_point(self.n_samples)
visualization.plot(points, space="M32")
self.KS.clear_points()
def test_draw_curve_pre_shape_2d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
times = gs.linspace(0.0, 1.0, 1000)
speeds = gs.array([-t * tangent_vec for t in times])
points = self.S32.ambient_metric.exp(speeds, base_point)
self.KS.add_points(points)
self.KS.draw_curve()
self.KS.clear_points()
def test_draw_vector_pre_shape_2d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
self.KS.draw_vector(tangent_vec, base_point)
def test_convert_to_spherical_coordinates_pre_shape_2d(self):
points = self.S32.random_point(self.n_samples)
coords = self.KS.convert_to_spherical_coordinates(points)
x = coords[:, 0]
y = coords[:, 1]
z = coords[:, 2]
result = x**2 + y**2 + z**2
expected = 0.25 * gs.ones(self.n_samples)
self.assertAllClose(result, expected)
def test_rotation_pre_shape_2d(self):
theta = gs.random.rand(1)[0]
phi = gs.random.rand(1)[0]
rot = self.KS.rotation(theta, phi)
result = _SpecialOrthogonalMatrices(3).belongs(rot)
expected = True
self.assertAllClose(result, expected)
def test_draw_pre_shape_3d(self):
self.KD.draw()
def test_draw_points_pre_shape_3d(self):
points = self.S33.random_point(self.n_samples)
visualization.plot(points, space="S33")
points = self.M33.random_point(self.n_samples)
visualization.plot(points, space="M33")
self.KD.clear_points()
def test_draw_curve_pre_shape_3d(self):
self.KD.draw()
base_point = self.S33.random_point()
vec = self.S33.random_point()
tangent_vec = self.S33.to_tangent(vec, base_point)
tangent_vec = 0.5 * tangent_vec / self.S33.ambient_metric.norm(
tangent_vec)
times = gs.linspace(0.0, 1.0, 1000)
speeds = gs.array([-t * tangent_vec for t in times])
points = self.S33.ambient_metric.exp(speeds, base_point)
self.KD.add_points(points)
self.KD.draw_curve()
self.KD.clear_points()
def test_draw_vector_pre_shape_3d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
self.KS.draw_vector(tangent_vec, base_point)
def test_convert_to_planar_coordinates_pre_shape_3d(self):
points = self.S33.random_point(self.n_samples)
coords = self.KD.convert_to_planar_coordinates(points)
x = coords[:, 0]
y = coords[:, 1]
radius = x**2 + y**2
result = [r <= 1.0 for r in radius]
self.assertTrue(gs.all(result))
@geomstats.tests.np_autograd_and_torch_only
def test_plot_points_s1(self):
points = self.S1.random_uniform(self.n_samples)
visualization.plot(points, space="S1")
def test_plot_points_s2(self):
points = self.S2.random_uniform(self.n_samples)
visualization.plot(points, space="S2")
def test_plot_points_h2_poincare_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space="H2_poincare_disk")
def test_plot_points_h2_poincare_half_plane_ext(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points,
space="H2_poincare_half_plane",
point_type="extrinsic")
def test_plot_points_h2_poincare_half_plane_none(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points, space="H2_poincare_half_plane")
def test_plot_points_h2_poincare_half_plane_hs(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points,
space="H2_poincare_half_plane",
point_type="half_space")
def test_plot_points_h2_klein_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space="H2_klein_disk")
@staticmethod
def test_plot_points_se2():
points = SpecialEuclidean(n=2, point_type="vector").random_point(4)
visu = visualization.SpecialEuclidean2(points, point_type="vector")
ax = visu.set_ax()
visu.draw_points(ax)
def test_plot_points_spd2(self):
one_point = self.spd.random_point()
visualization.plot(one_point, space="SPD2")
points = self.spd.random_point(4)
visualization.plot(points, space="SPD2")
def test_compute_coordinates_spd2(self):
point = gs.eye(2)
ellipsis = visualization.Ellipses(n_sampling_points=4)
x, y = ellipsis.compute_coordinates(point)
self.assertAllClose(x, gs.array([1, 0, -1, 0, 1]))
self.assertAllClose(y, gs.array([0, 1, 0, -1, 0]))
@staticmethod
def teardown_method():
plt.close()
class TestVisualization(geomstats.tests.TestCase):
def setUp(self):
self.n_samples = 10
self.SO3_GROUP = SpecialOrthogonal(n=3, point_type='vector')
self.SE3_GROUP = SpecialEuclidean(n=3, point_type='vector')
self.S1 = Hypersphere(dim=1)
self.S2 = Hypersphere(dim=2)
self.H2 = Hyperbolic(dim=2)
self.H2_half_plane = PoincareHalfSpace(dim=2)
self.M32 = Matrices(m=3, n=2)
self.S32 = PreShapeSpace(k_landmarks=3, m_ambient=2)
self.KS = visualization.KendallSphere()
self.M33 = Matrices(m=3, n=3)
self.S33 = PreShapeSpace(k_landmarks=3, m_ambient=3)
self.KD = visualization.KendallDisk()
plt.figure()
@staticmethod
def test_tutorial_matplotlib():
visualization.tutorial_matplotlib()
def test_plot_points_so3(self):
points = self.SO3_GROUP.random_uniform(self.n_samples)
visualization.plot(points, space='SO3_GROUP')
def test_plot_points_se3(self):
points = self.SE3_GROUP.random_point(self.n_samples)
visualization.plot(points, space='SE3_GROUP')
def test_draw_pre_shape_2d(self):
self.KS.draw()
def test_draw_points_pre_shape_2d(self):
points = self.S32.random_point(self.n_samples)
visualization.plot(points, space='S32')
points = self.M32.random_point(self.n_samples)
visualization.plot(points, space='M32')
self.KS.clear_points()
def test_draw_curve_pre_shape_2d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
times = gs.linspace(0., 1., 1000)
speeds = gs.array([-t * tangent_vec for t in times])
points = self.S32.ambient_metric.exp(speeds, base_point)
self.KS.add_points(points)
self.KS.draw_curve()
self.KS.clear_points()
def test_draw_vector_pre_shape_2d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
self.KS.draw_vector(tangent_vec, base_point)
def test_convert_to_spherical_coordinates_pre_shape_2d(self):
points = self.S32.random_point(self.n_samples)
coords = self.KS.convert_to_spherical_coordinates(points)
x = coords[:, 0]
y = coords[:, 1]
z = coords[:, 2]
result = x**2 + y**2 + z**2
expected = .25 * gs.ones(self.n_samples)
self.assertAllClose(result, expected)
def test_rotation_pre_shape_2d(self):
theta = gs.random.rand(1)[0]
phi = gs.random.rand(1)[0]
rot = self.KS.rotation(theta, phi)
result = _SpecialOrthogonalMatrices(3).belongs(rot)
expected = True
self.assertAllClose(result, expected)
def test_draw_pre_shape_3d(self):
self.KD.draw()
def test_draw_points_pre_shape_3d(self):
points = self.S33.random_point(self.n_samples)
visualization.plot(points, space='S33')
points = self.M33.random_point(self.n_samples)
visualization.plot(points, space='M33')
self.KD.clear_points()
def test_draw_curve_pre_shape_3d(self):
self.KD.draw()
base_point = self.S33.random_point()
vec = self.S33.random_point()
tangent_vec = self.S33.to_tangent(vec, base_point)
tangent_vec = .5 * tangent_vec / self.S33.ambient_metric.norm(
tangent_vec)
times = gs.linspace(0., 1., 1000)
speeds = gs.array([-t * tangent_vec for t in times])
points = self.S33.ambient_metric.exp(speeds, base_point)
self.KD.add_points(points)
self.KD.draw_curve()
self.KD.clear_points()
def test_draw_vector_pre_shape_3d(self):
self.KS.draw()
base_point = self.S32.random_point()
vec = self.S32.random_point()
tangent_vec = self.S32.to_tangent(vec, base_point)
self.KS.draw_vector(tangent_vec, base_point)
def test_convert_to_planar_coordinates_pre_shape_3d(self):
points = self.S33.random_point(self.n_samples)
coords = self.KD.convert_to_planar_coordinates(points)
x = coords[:, 0]
y = coords[:, 1]
radius = x**2 + y**2
result = [r <= 1. for r in radius]
self.assertTrue(gs.all(result))
@geomstats.tests.np_and_pytorch_only
def test_plot_points_s1(self):
points = self.S1.random_uniform(self.n_samples)
visualization.plot(points, space='S1')
def test_plot_points_s2(self):
points = self.S2.random_uniform(self.n_samples)
visualization.plot(points, space='S2')
def test_plot_points_h2_poincare_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space='H2_poincare_disk')
def test_plot_points_h2_poincare_half_plane_ext(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points,
space='H2_poincare_half_plane',
point_type='extrinsic')
def test_plot_points_h2_poincare_half_plane_none(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points, space='H2_poincare_half_plane')
def test_plot_points_h2_poincare_half_plane_hs(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points,
space='H2_poincare_half_plane',
point_type='half_space')
def test_plot_points_h2_klein_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space='H2_klein_disk')
@staticmethod
def test_plot_points_se2():
points = SpecialEuclidean(n=2, point_type='vector').random_point(4)
visu = visualization.SpecialEuclidean2(points, point_type='vector')
ax = visu.set_ax()
visu.draw(ax)
class TestVisualization(geomstats.tests.TestCase):
def setUp(self):
self.n_samples = 10
self.SO3_GROUP = SpecialOrthogonal(n=3, point_type='vector')
self.SE3_GROUP = SpecialEuclidean(n=3, point_type='vector')
self.S1 = Hypersphere(dim=1)
self.S2 = Hypersphere(dim=2)
self.H2 = Hyperbolic(dim=2)
self.H2_half_plane = PoincareHalfSpace(dim=2)
plt.figure()
@staticmethod
def test_tutorial_matplotlib():
visualization.tutorial_matplotlib()
def test_plot_points_so3(self):
points = self.SO3_GROUP.random_uniform(self.n_samples)
visualization.plot(points, space='SO3_GROUP')
def test_plot_points_se3(self):
points = self.SE3_GROUP.random_point(self.n_samples)
visualization.plot(points, space='SE3_GROUP')
@geomstats.tests.np_and_pytorch_only
def test_plot_points_s1(self):
points = self.S1.random_uniform(self.n_samples)
visualization.plot(points, space='S1')
def test_plot_points_s2(self):
points = self.S2.random_uniform(self.n_samples)
visualization.plot(points, space='S2')
def test_plot_points_h2_poincare_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space='H2_poincare_disk')
def test_plot_points_h2_poincare_half_plane_ext(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points,
space='H2_poincare_half_plane',
point_type='extrinsic')
def test_plot_points_h2_poincare_half_plane_none(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points, space='H2_poincare_half_plane')
def test_plot_points_h2_poincare_half_plane_hs(self):
points = self.H2_half_plane.random_point(self.n_samples)
visualization.plot(points,
space='H2_poincare_half_plane',
point_type='half_space')
def test_plot_points_h2_klein_disk(self):
points = self.H2.random_point(self.n_samples)
visualization.plot(points, space='H2_klein_disk')
@staticmethod
def test_plot_points_se2():
points = SpecialEuclidean(n=2, point_type='vector').random_point(4)
visu = visualization.SpecialEuclidean2(points, point_type='vector')
ax = visu.set_ax()
visu.draw(ax)