Add checks in simulation test that ensure rewards and returns vectors…

… have size equal to the number of players. PiperOrigin-RevId: 272283762 Change-Id: I8aff35e4f8093fbd15619392ce35c38c296b1fa0
ifeela · Oct 3, 2019 · e0f6cb5 · e0f6cb5
1 parent 91ff14b
commit e0f6cb5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/open_spiel/tests/basic_tests.cc b/open_spiel/tests/basic_tests.cc
@@ -335,6 +335,7 @@ void RandomSimulation(std::mt19937* rng, const Game& game, bool undo) {
       }
     } else if (state->CurrentPlayer() == open_spiel::kSimultaneousPlayerId) {
       std::vector<double> rewards = state->Rewards();
+      SPIEL_CHECK_EQ(rewards.size(), game.NumPlayers());
       std::cout << "Rewards: " << absl::StrJoin(rewards, " ") << std::endl;
       for (auto p = Player{0}; p < game.NumPlayers(); ++p) {
         episode_returns[p] += rewards[p];
@@ -370,6 +371,7 @@ void RandomSimulation(std::mt19937* rng, const Game& game, bool undo) {
       game_length++;
     } else {
       std::vector<double> rewards = state->Rewards();
+      SPIEL_CHECK_EQ(rewards.size(), game.NumPlayers());
       std::cout << "Rewards: " << absl::StrJoin(rewards, " ") << std::endl;
       for (auto p = Player{0}; p < game.NumPlayers(); ++p) {
         episode_returns[p] += rewards[p];
@@ -432,6 +434,7 @@ void RandomSimulation(std::mt19937* rng, const Game& game, bool undo) {
   }
 
   auto returns = state->Returns();
+  SPIEL_CHECK_EQ(returns.size(), game.NumPlayers());
   for (Player player = 0; player < game.NumPlayers(); player++) {
     double final_return = returns[player];
     SPIEL_CHECK_FLOAT_EQ(final_return, state->PlayerReturn(player));