From 999ca3d6c541160d9800f27a9d71b7c4d3c48a36 Mon Sep 17 00:00:00 2001
From: Andrey Vlasovskikh <andrey.vlasovskikh@jetbrains.com>
Date: Tue, 31 Jan 2012 16:25:43 +0400
Subject: [PATCH] Fixed AE in PyKeywordArgumentReference (PY-5586)

---
 .../python/psi/impl/PyKeywordArgumentReference.java       | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/python/src/com/jetbrains/python/psi/impl/PyKeywordArgumentReference.java b/python/src/com/jetbrains/python/psi/impl/PyKeywordArgumentReference.java
index f23e9676d4ff..c25e27029561 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyKeywordArgumentReference.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyKeywordArgumentReference.java
@@ -22,9 +22,11 @@ public class PyKeywordArgumentReference extends PsiReferenceBase.Poly<PyKeywordA
   @NotNull
   @Override
   public ResolveResult[] multiResolve(boolean incompleteCode) {
-    PyCallExpression call = PsiTreeUtil.getParentOfType(myElement, PyCallExpression.class);
-    assert call != null;
-    final PyExpression callee = call.getCallee();
+    PsiElement call = PsiTreeUtil.getParentOfType(myElement, PyCallExpression.class, PyClass.class);
+    if (!(call instanceof PyCallExpression)) {
+      return new ResolveResult[0];
+    }
+    final PyExpression callee = ((PyCallExpression)call).getCallee();
     if (callee == null) return ResolveResult.EMPTY_ARRAY;
     final PsiPolyVariantReference calleeReference = (PsiPolyVariantReference) callee.getReference();
     if (calleeReference == null) return ResolveResult.EMPTY_ARRAY;