def test_elevon2aileron_elevator():
control_input = ControlInput()
test_vec1 = np.zeros(2)
control_input.control_input[:2] = test_vec1
val1 = control_input.elevon2aileron_elevator(test_vec1)
test_vec2 = np.ones(2)
control_input.control_input[:2] = test_vec2
val2 = control_input.elevon2aileron_elevator(test_vec2)
expect1 = np.zeros(2)
expect2 = np.array([1, 0])
assert np.allclose(val1, expect1)
assert np.allclose(val2, expect2)
def test_flying_wing_setters():
control_input = ControlInput()
throttle = 1
elevon_left = 2
elevon_right = 1
elevon_vec = np.array([elevon_right, elevon_left])
control_input.elevon_left = elevon_left
control_input.elevon_right = elevon_right
control_input.throttle = throttle
expect_throttle = 1
expect_elev_ail = control_input.elevon2aileron_elevator(elevon_vec)
expect_control_input = np.array(
[expect_elev_ail[0], expect_elev_ail[1], 0, expect_throttle])
assert np.allclose(control_input.control_input, expect_control_input)
