namespace Gen7 {
    [Ability effect=ArenaTrap]
    class ArenaTrap : PkmnScript {
        void PreventOpponentRunAway(FleeTurnChoice@ choice, bool &inout block){
            if (block) return;
            if (SpecialStates::IsRaised(choice.User)) return;
            block = true;
        }

        void PreventOpponentSwitch(SwitchTurnChoice@ choice, bool &inout block){
            if (block) return;
            if (SpecialStates::IsRaised(choice.User)) return;
            block = true;
        }
    }
}