List<PyStatement> (@NotNull final PyFile pyFile) { final TypeEvalContext context = TypeEvalContext.userInitiated(pyFile.getProject(), pyFile); return pyFile.getTopLevelClasses().stream() .filter(o -> PythonUnitTestDetectorsBasedOnSettings.isTestClass(o, ThreeState.UNSURE, context)) .collect(Collectors.toList()); }