def test_set_and_vector_transform(self):
state = HeightState(height_above_ground, vertical_speed, ground_height_barometer, ground_height_gps)
self.assertEqual(state.height_above_ground, height_above_ground)
self.assertEqual(state.vertical_speed, vertical_speed)
self.assertEqual(state.ground_height_barometer, ground_height_barometer)
self.assertEqual(state.ground_height_gps, ground_height_gps)
vector = state.as_vector()
self.assertEqual(vector[0], state.height_above_ground)
self.assertEqual(vector[1], state.vertical_speed)
self.assertEqual(vector[2], state.ground_height_barometer)
self.assertEqual(vector[3], state.ground_height_gps)
def test_set_from_vector2(self):
vector = array([[height_above_ground], [vertical_speed], [ground_height_barometer], [ground_height_gps]])
state = HeightState.fromVector(vector)
self.assertEqual(state.height_above_ground, height_above_ground)
self.assertEqual(state.vertical_speed, vertical_speed)
self.assertEqual(state.ground_height_barometer, ground_height_barometer)
self.assertEqual(state.ground_height_gps, ground_height_gps)