def test_basis_representation_is_correctly_vectorized(self):
for n in range(2, 5):
algebra = SpecialEuclideanMatrixLieAlgebra(n)
shape = gs.shape(algebra.basis_representation(algebra.basis))
dim = int(n * (n + 1) / 2)
self.assertAllClose(shape, (dim, dim))
def test_basis_has_the_right_dimension(self):
for n in range(2, 5):
algebra = SpecialEuclideanMatrixLieAlgebra(n)
self.assertEqual(int(n * (n + 1) / 2), algebra.dim)