def test_rot(): V = [1, 0, 0, 1] rotated_V = proj3d.rot_x(V, np.pi / 6) np.testing.assert_allclose(rotated_V, [1, 0, 0, 1]) V = [0, 1, 0, 1] rotated_V = proj3d.rot_x(V, np.pi / 6) np.testing.assert_allclose(rotated_V, [0, np.sqrt(3) / 2, 0.5, 1])