def test_cylindrical_setpoints(current_driver, set_target):
"""
Check that the individual x, y, z instruments are getting the set points as intended. We can do this because the
instruments are printing log messages to an IO stream. We intercept these messages and extract the log lines which
mention that the instrument is ramping to certain values. These values should match the values of the input. In
this test we are verifying this for cylindrical coordinates.
"""
current_driver.cylindrical(set_target)
reported_ramp_targets = get_instruments_ramp_messages(current_driver)
get_target = {k: v["value"] for k, v in reported_ramp_targets.items()}
set_vector = FieldVector(**dict(zip(["rho", "phi", "z"], set_target)))
get_vector = FieldVector(**get_target)
assert set_vector.is_equal(get_vector)
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)