PyLoopStatement (@NotNull PsiElement breakOrContinue) { return breakOrContinue instanceof PyContinueStatement || breakOrContinue instanceof PyBreakStatement ? getCorrespondingLoopImpl(breakOrContinue) : null; }