List<PsiElement> () { var file = (PyFile)myFixture.configureByFile(FOLDER_NAME + "/test_test.py"); var clazz = file.findTopLevelClass("TestFoo"); assert clazz != null : "No test found"; return Collections.singletonList(clazz); }