def setUp(self):
"""setUp method.
Use the setUp method to define variables that remain constant
during all tests. For example, here we test the
4-dimensional manifold of the class MyManifold.
"""
self.dimension = 4
self.another_parameter = 3
self.manifold = MyManifold(dim=self.dimension, another_parameter=3)
class TestMyManifold(geomstats.tests.TestCase):
"""Class testing the methods of MyManifold.
In the class TestMyManifold, each test method:
- needs to start with `test_`
- represents a unit-test, i.e. it tests one and only one method
or attribute of the class MyManifold,
- ends with the line: `self.assertAllClose(result, expected)`,
as in the examples below.
"""
def setUp(self):
"""setUp method.
Use the setUp method to define variables that remain constant
during all tests. For example, here we test the
4-dimensional manifold of the class MyManifold.
"""
self.dimension = 4
self.another_parameter = 3
self.manifold = MyManifold(dim=self.dimension, another_parameter=3)
def test_dimension(self):
"""Test dimension.
The method test_dimension tests the `dim` attribute.
"""
result = self.manifold.dim
expected = self.dimension
# Each test ends with the following syntax, comparing
# the result with the expected result, using self.assertAllClose
self.assertAllClose(result, expected)
def test_belongs(self):
"""Test belongs.
The method test_belongs tests the `belongs` method.
Note that arrays are defined using geomstats backend
through the prefix `gs.`.
This allows the code to be tested simultaneously in numpy,
pytorch and tensorflow. `gs.` is the equivalent of numpy's `np.` and
most of numpy's functions are available with `gs.`.
"""
point = gs.array([1.0, 2.0, 3.0])
result = self.manifold.belongs(point)
expected = False
self.assertAllClose(result, expected)
def test_belongs_vectorization(self):
"""Test belongs with several input points.
All functions and methods should work with several input points,
or vectors.
"""
point = gs.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
result = self.manifold.belongs(point)
expected = gs.array([False, False])
self.assertAllClose(result, expected)
def test_is_tangent(self):
"""Test is_tangent.
The method test_is_tangent tests the `is_tangent` method.
Note that arrays are defined using geomstats backend
through the prefix `gs.`.
This allows the code to be tested simultaneously in numpy,
pytorch and tensorflow. `gs.` is the equivalent of numpy's `np.` and
most of numpy's functions are available with `gs.`.
"""
vector = gs.array([1.0, 2.0, 3.0, 4.0])
result = self.manifold.is_tangent(vector)
expected = True
self.assertAllClose(result, expected)
def test_is_tangent_vectorization(self):
"""Test is_tangent with several input vectors
All functions and methods should work with several input points,
or vectors.
"""
vector = gs.array([[1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0]])
result = self.manifold.is_tangent(vector)
expected = gs.array([True, True])
self.assertAllClose(result, expected)