diff --git a/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt b/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt index 8edc5eeaf7..46f4a7433c 100644 --- a/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt +++ b/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt @@ -8,6 +8,7 @@ import org.utbot.python.framework.codegen.model.PythonUserImport import org.utbot.framework.codegen.domain.TestFramework import org.utbot.framework.codegen.domain.models.CgMethodTestSet import org.utbot.framework.plugin.api.ExecutableId +import org.utbot.framework.plugin.api.UtClusterInfo import org.utbot.framework.plugin.api.UtExecutionSuccess import org.utbot.framework.plugin.api.util.UtContext import org.utbot.framework.plugin.api.util.withUtContext @@ -200,9 +201,12 @@ object PythonTestGenerationProcessor { ) val testCode = codegen.pythonGenerateAsStringWithTestReport( notEmptyTests.map { testSet -> + val intRange = testSet.executions.indices + val clusterInfo = listOf(Pair(UtClusterInfo("FUZZER"), intRange)) CgMethodTestSet( executableId = methodIds[testSet.method] as ExecutableId, - executions = testSet.executions + executions = testSet.executions, + clustersInfo = clusterInfo, ) }, allImports @@ -299,4 +303,4 @@ object PythonTestGenerationProcessor { } else { paths } -} \ No newline at end of file +}