def test_inverse_rotation():
"""Test the inverse rotation."""
xyz = np.concatenate((np.ones((2, 4, 1)), np.zeros(
(2, 4, 1)), np.zeros((2, 4, 1))), -1)
c = Coordinates(xyz[..., 0].copy(), xyz[..., 1].copy(), xyz[..., 2].copy())
c.rotate('z', 90)
c.rotate('z', 90, inverse=True)
npt.assert_allclose(c.get_cart(), xyz, atol=1e-15)
def test_rotation_assertion():
"""Test rotation with unknown rotation type."""
c = Coordinates(1, 0, 0)
# test with unknown type
with raises(ValueError):
c.rotate('urgh', 90)
def test_rotation(rot_type, rot):
"""Test rotation with different formats."""
c = Coordinates(1, 0, 0)
c.rotate(rot_type, rot)
npt.assert_allclose(c.get_cart(), np.atleast_2d([0, 1, 0]), atol=1e-15)