void () { final PyFile file = getTestFile(); final List<PyClass> classes = file.getTopLevelClasses(); assertEquals(1, classes.size()); PyClass pyClass = classes.get(0); assertEquals("SomeClass", pyClass.getName()); assertNotParsed(file); // load the tree now final PyStatementList statements = pyClass.getStatementList(); assertNotNull(((PyFileImpl)file).getTreeElement()); final PsiElement[] children = file.getChildren(); assertEquals(1, children.length); assertSame(pyClass, children[0]); }