if n == 3:
point = self.elements[3]["with_angle_0"]
self.assertTrue(gs.isclose(gs.linalg.norm(point), 0.))
result = group.regularize(point)
expected = point
self.assertTrue(gs.allclose(result, expected), "! angle 0 !")
After Change
self.assertAllClose(result, expected)
else:
angle = 0.345
point = gs.array([[
[gs.cos(angle), -gs.sin(angle)],
[gs.sin(angle), gs.cos(angle)]]])
result = group.regularize(point)
expected = point
self.assertAllClose(result, expected)