diff --git a/modules/prediction/predictor/pedestrian/regional_predictor_test.cc b/modules/prediction/predictor/pedestrian/regional_predictor_test.cc index d06e6eb96541d448795c24783e2134d32e0826a4..41e0f318fff341b61b483a25f8e3ec0ca91e6b46 100644 --- a/modules/prediction/predictor/pedestrian/regional_predictor_test.cc +++ b/modules/prediction/predictor/pedestrian/regional_predictor_test.cc @@ -38,6 +38,9 @@ class RegionalPredictorTest : public ::testing::Test { "modules/prediction/testdata/multiple_perception_pedestrians.pb.txt"; apollo::common::util::GetProtoFromFile(file, &perception_obstacles_); FLAGS_map_file = "modules/prediction/testdata/kml_map.bin"; + FLAGS_p_var = 0.1; + FLAGS_q_var = 0.01; + FLAGS_r_var = 0.25; } protected: apollo::perception::PerceptionObstacles perception_obstacles_; diff --git a/modules/prediction/predictor/vehicle/free_move_predictor_test.cc b/modules/prediction/predictor/vehicle/free_move_predictor_test.cc index e69a0c1c10b2ecefb1ebdffe719cfdca1bfaf84a..787749bf20db7292bdeab24cb3cbd0885b94cce0 100644 --- a/modules/prediction/predictor/vehicle/free_move_predictor_test.cc +++ b/modules/prediction/predictor/vehicle/free_move_predictor_test.cc @@ -38,12 +38,35 @@ class FreeMovePredictorTest : public ::testing::Test { "modules/prediction/testdata/single_perception_vehicle_offlane.pb.txt"; apollo::common::util::GetProtoFromFile(file, &perception_obstacles_); FLAGS_map_file = "modules/prediction/testdata/kml_map.bin"; + FLAGS_p_var = 0.1; + FLAGS_q_var = 0.01; + FLAGS_r_var = 0.25; } protected: apollo::perception::PerceptionObstacles perception_obstacles_; }; TEST_F(FreeMovePredictorTest, General) { + EXPECT_DOUBLE_EQ(perception_obstacles_.header().timestamp_sec(), + 1501183430.161906); + apollo::perception::PerceptionObstacle perception_obstacle = + perception_obstacles_.perception_obstacle(0); + EXPECT_EQ(perception_obstacle.id(), 15); + ObstaclesContainer container; + container.Insert(perception_obstacles_); + Obstacle* obstacle_ptr = container.GetObstacle(15); + EXPECT_TRUE(obstacle_ptr != nullptr); + FreeMovePredictor predictor; + predictor.Predict(obstacle_ptr); + const PredictionObstacle& prediction_obstacle = + predictor.prediction_obstacle(); + EXPECT_EQ(prediction_obstacle.trajectory_size(), 1); + EXPECT_NEAR( + prediction_obstacle.trajectory(0).trajectory_point(9).path_point().x(), + -432.459, 0.001); + EXPECT_NEAR( + prediction_obstacle.trajectory(0).trajectory_point(9).path_point().y(), + -156.451, 0.001); } } // namespace prediction diff --git a/modules/prediction/testdata/single_perception_vehicle_offlane.pb.txt b/modules/prediction/testdata/single_perception_vehicle_offlane.pb.txt index 2766b12bf471092e1e168ca6f074c68130d95995..263884674032581334b6c747bcfcf89c032fe0b2 100644 --- a/modules/prediction/testdata/single_perception_vehicle_offlane.pb.txt +++ b/modules/prediction/testdata/single_perception_vehicle_offlane.pb.txt @@ -4,7 +4,7 @@ header { sequence_num: 15839 } perception_obstacle { - id: 1 + id: 15 position { x: -438.614, y: -173.366,