From 5e314dbb48eee166a89894527dee2ce6ba797deb Mon Sep 17 00:00:00 2001
From: Andrey Vlasovskikh <andrey.vlasovskikh@gmail.com>
Date: Thu, 30 Aug 2012 16:15:13 +0400
Subject: [PATCH] Added runWithLanguageLevel() for Python testes

---
 .../com/jetbrains/python/fixtures/PyTestCase.java     | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java
index f1142c26a00d..6697b0775462 100644
--- a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java
+++ b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java
@@ -27,6 +27,7 @@ import com.jetbrains.python.psi.LanguageLevel;
 import com.jetbrains.python.psi.PyFile;
 import com.jetbrains.python.psi.impl.PyFileImpl;
 import com.jetbrains.python.psi.impl.PythonLanguageLevelPusher;
+import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
 /**
@@ -89,6 +90,16 @@ public abstract class PyTestCase extends UsefulTestCase {
     PythonLanguageLevelPusher.setForcedLanguageLevel(myFixture.getProject(), languageLevel);
   }
 
+  protected void runWithLanguageLevel(@NotNull LanguageLevel languageLevel, @NotNull Runnable action) {
+    setLanguageLevel(languageLevel);
+    try {
+      action.run();
+    }
+    finally {
+      setLanguageLevel(null);
+    }
+  }
+
   protected static void assertNotParsed(PyFile file) {
     assertNull(PARSED_ERROR_MSG, ((PyFileImpl)file).getTreeElement());
   }