From 64cc818142125db85357d6de2f25da9b96ebc962 Mon Sep 17 00:00:00 2001
From: Ekaterina Tuzova <Ekaterina.Tuzova@jetbrains.com>
Date: Mon, 10 Jun 2013 15:35:43 +0400
Subject: [PATCH] do not search for init in test classes

---
 python/psi-api/src/com/jetbrains/python/PyNames.java            | 2 ++
 .../python/inspections/PyClassHasNoInitInspection.java          | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/python/psi-api/src/com/jetbrains/python/PyNames.java b/python/psi-api/src/com/jetbrains/python/PyNames.java
index 03f116dfcd76..421a020e4209 100644
--- a/python/psi-api/src/com/jetbrains/python/PyNames.java
+++ b/python/psi-api/src/com/jetbrains/python/PyNames.java
@@ -128,6 +128,8 @@ public class PyNames {
   public static final String AT_TEST = "Attest";
   public static final String AT_TEST_IMPORT = "attest";
 
+  public static final String TEST_CASE = "TestCase";
+
   public static final String PYCACHE =  "__pycache__";
 
   public static final String NOT_IMPLEMENTED_ERROR = "NotImplementedError";
diff --git a/python/src/com/jetbrains/python/inspections/PyClassHasNoInitInspection.java b/python/src/com/jetbrains/python/inspections/PyClassHasNoInitInspection.java
index 9ab859bc4a9e..c7dfb09a8512 100644
--- a/python/src/com/jetbrains/python/inspections/PyClassHasNoInitInspection.java
+++ b/python/src/com/jetbrains/python/inspections/PyClassHasNoInitInspection.java
@@ -5,6 +5,7 @@ import com.intellij.codeInspection.ProblemsHolder;
 import com.intellij.psi.PsiElementVisitor;
 import com.intellij.psi.PsiReference;
 import com.jetbrains.python.PyBundle;
+import com.jetbrains.python.PyNames;
 import com.jetbrains.python.inspections.quickfix.AddMethodQuickFix;
 import com.jetbrains.python.psi.PyClass;
 import com.jetbrains.python.psi.PyExpression;
@@ -45,6 +46,7 @@ public class PyClassHasNoInitInspection extends PyInspection {
       for (PyExpression pyClass : classes) {
         final PsiReference reference = pyClass.getReference();
         if (reference == null || reference.resolve() == null) return;
+        if (reference.getCanonicalText().contains(PyNames.TEST_CASE)) return;
       }
       final PyFunction init = node.findInitOrNew(true);
       if (init == null) {