diff options
Diffstat (limited to 'python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java')
-rw-r--r-- | python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java index 3404194efd65..6850c3b049a8 100644 --- a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java +++ b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java @@ -16,13 +16,14 @@ package com.jetbrains.python.fixtures; import com.intellij.openapi.application.PathManager; -import com.intellij.openapi.module.EmptyModuleType; +import com.intellij.openapi.extensions.Extensions; import com.intellij.openapi.module.Module; import com.intellij.openapi.module.ModuleType; import com.intellij.openapi.projectRoots.Sdk; import com.intellij.openapi.roots.ContentEntry; import com.intellij.openapi.roots.ModifiableRootModel; import com.intellij.openapi.roots.OrderRootType; +import com.intellij.openapi.roots.impl.FilePropertyPusher; import com.intellij.openapi.roots.libraries.Library; import com.intellij.openapi.vfs.LocalFileSystem; import com.intellij.openapi.vfs.VirtualFile; @@ -39,6 +40,7 @@ import com.intellij.testFramework.fixtures.TestFixtureBuilder; import com.intellij.testFramework.fixtures.impl.LightTempDirTestFixtureImpl; import com.jetbrains.python.PythonHelpersLocator; import com.jetbrains.python.PythonMockSdk; +import com.jetbrains.python.PythonModuleTypeBase; import com.jetbrains.python.PythonTestUtil; import com.jetbrains.python.psi.LanguageLevel; import com.jetbrains.python.psi.PyClass; @@ -92,6 +94,8 @@ public abstract class PyTestCase extends UsefulTestCase { setLanguageLevel(null); myFixture.tearDown(); myFixture = null; + final PythonLanguageLevelPusher levelPusher = Extensions.findExtension(FilePropertyPusher.EP_NAME, PythonLanguageLevelPusher.class); + levelPusher.flushLanguageLevelCache(); super.tearDown(); } @@ -149,7 +153,7 @@ public abstract class PyTestCase extends UsefulTestCase { @Override public ModuleType getModuleType() { - return EmptyModuleType.getInstance(); + return PythonModuleTypeBase.getInstance(); } @Override |