List<PsiElement> () { final List<PsiElement> result = new ArrayList<>(); result.add(getDir("tests_package/package_test")); result.add(getFile("tests_package/package_test", "test_in_package.py")); result.add(getFile("/", "test_foo.py")); result.add(getFile("tests_package/package_test", "test_in_package.py").findTopLevelClass("TestLogic")); result.add(getDir("tests_folder")); result.add(getFile("tests_folder", "test_lonely.py")); result.add(getFile("tests_folder", "test_lonely.py").findTopLevelClass("TestLonely")); result.add(getFile("tests_folder", "test_functions.py").findTopLevelFunction("test_test")); result.add(getFile("tests_folder", "test_functions.py").findTopLevelFunction("foo")); result.add(getFile("tests_folder/test-test", "test-foo.py").findTopLevelClass("TestDash")); return result; }