def test_cartesian_setpoints(current_driver, set_target): """ Check that the individual x, y, z instruments are getting the set points as intended. This test is very similar to the sanity test, but adds in the FieldVector as well. """ current_driver.cartesian(set_target) x = current_driver.x() y = current_driver.y() z = current_driver.z() get_target = dict(zip(('x', 'y', 'z'), (x, y, z))) set_vector = FieldVector(*set_target) get_vector = FieldVector(**get_target) assert set_vector.is_equal(get_vector)
def test_cylindrical_setpoints(current_driver, set_target): """ Check that the individual x, y, z instruments are getting the set points as intended. This test is very similar to the sanity test, but adds in the FieldVector as well. """ current_driver.cylindrical(set_target) rho = current_driver.rho() z = current_driver.z() phi = current_driver.phi() get_target = dict(zip(('rho', 'phi', 'z'), (rho, phi, z))) set_target = dict(zip(('rho', 'phi', 'z'), set_target)) set_vector = FieldVector(**set_target) get_vector = FieldVector(**get_target) assert set_vector.is_equal(get_vector)
def test_spherical_setpoints(current_driver, set_target): """ Check that the individual x, y, z instruments are getting the set points as intended. This test is very similar to the sanity test, but adds in the FieldVector as well. """ current_driver.spherical(set_target) r = current_driver.field() theta = current_driver.theta() phi = current_driver.phi() get_target = dict(zip(('r', 'theta', 'phi'), (r, theta, phi))) set_target = dict(zip(('r', 'theta', 'phi'), set_target)) set_vector = FieldVector(**set_target) get_vector = FieldVector(**get_target) assert set_vector.is_equal(get_vector)