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)