def test_cartesian_to_spherical_one_4D_point(self):
x_cart = (1., 1., 0., 0.)
y_cart = (1., 0., 1., 0.)
z_cart = (0., 0., 0., 1.)
x_conv = sp.cartesian_to_spherical(x_cart)
y_conv = sp.cartesian_to_spherical(y_cart)
z_conv = sp.cartesian_to_spherical(z_cart)
x_spherical = (1., 1., pi / 2, 0.)
y_spherical = (1., 1., pi / 2, pi / 2)
z_spherical = (0., 1., 0., 0.)
self.assertTrue(
np.array_equal(np.round(x_conv, 7), np.round(x_spherical, 7)))
self.assertTrue(
np.array_equal(np.round(y_conv, 7), np.round(y_spherical, 7)))
self.assertTrue(
np.array_equal(np.round(z_conv, 7), np.round(z_spherical, 7)))
def test_cartesian_to_spherical_one_3D_complex_point(self):
x_iy_cart = (1., 1.j, 0.)
x_iy_conv = sp.cartesian_to_spherical(x_iy_cart)
x_iy_spherical = (1. + 1.j, pi / 2 + 1.j * pi / 2, 1.j * pi / 2)
self.assertTrue(
np.array_equal(np.round(x_iy_conv, 7), np.round(x_iy_spherical,
7)))
def matrix_to_vector(matrix):
cart_vec = calculate_cartesian_spin_vector_from_matrix(matrix)
return sp.cartesian_to_spherical(cart_vec)