Method OnStep

tests/test_unit_command_common.cc:25–42

    }

    void TestUnitCommand::OnStep() {
        const ObservationInterface* obs = agent_->Observation();
        ActionInterface* act = agent_->Actions();

        if (obs->GetGameLoop() < order_on_game_loop_) {
            return;
        }

        test_units_ = obs->GetUnits(Unit::Self, [&](const Unit& unit) {
            return unit.unit_type == test_unit_type_;
        });

        if (test_units_.size() > 0) {
            test_unit_ = test_units_.front();
        }

        IssueUnitCommand(act);
    }

    void TestUnitCommand::OnTestFinish() {
        VerifyUnitIdleAfterOrder(test_unit_type_);