diff --git a/python/testData/stubs/ImportFeatures.py b/python/testData/stubs/ImportFeatures.py new file mode 100644 index 000000000000..ca4872b8365f --- /dev/null +++ b/python/testData/stubs/ImportFeatures.py @@ -0,0 +1 @@ +from __future__ import unicode_literals, division diff --git a/python/testSrc/com/jetbrains/python/PyStubsTest.java b/python/testSrc/com/jetbrains/python/PyStubsTest.java index cc4e961eb4aa..6d165f5875c0 100644 --- a/python/testSrc/com/jetbrains/python/PyStubsTest.java +++ b/python/testSrc/com/jetbrains/python/PyStubsTest.java @@ -296,6 +296,16 @@ public class PyStubsTest extends PyLightFixtureTestCase { assertNotParsed(file); } + + public void testImportFeatures() { + final PyFileImpl file = (PyFileImpl) getTestFile(); + assertTrue(file.hasImportFromFuture(FutureFeature.DIVISION)); + assertTrue(file.hasImportFromFuture(FutureFeature.UNICODE_LITERALS)); + assertNotParsed(file); + } + + // --- + private PyFile getTestFile() { return getTestFile(getTestName(false) + ".py"); }