PyMemberInfoStorage (PyClass clazz, PsiElement element1, PsiElement element2) { final PyMemberInfoStorage infoStorage = new PyMemberInfoStorage(clazz); for (PyMemberInfo<PyElement> member : infoStorage.getClassMemberInfos(clazz)) { final PyElement function = member.getMember(); member.setChecked(PsiTreeUtil.isAncestor(function, element1, false) || PsiTreeUtil.isAncestor(function, element2, false)); } return infoStorage; }