Skip to content

Commit aff18de

Browse files
authored
Add #region name for generated Python tests (#1853)
1 parent 2535296 commit aff18de

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import org.utbot.python.framework.codegen.model.PythonUserImport
88
import org.utbot.framework.codegen.domain.TestFramework
99
import org.utbot.framework.codegen.domain.models.CgMethodTestSet
1010
import org.utbot.framework.plugin.api.ExecutableId
11+
import org.utbot.framework.plugin.api.UtClusterInfo
1112
import org.utbot.framework.plugin.api.UtExecutionSuccess
1213
import org.utbot.framework.plugin.api.util.UtContext
1314
import org.utbot.framework.plugin.api.util.withUtContext
@@ -200,9 +201,12 @@ object PythonTestGenerationProcessor {
200201
)
201202
val testCode = codegen.pythonGenerateAsStringWithTestReport(
202203
notEmptyTests.map { testSet ->
204+
val intRange = testSet.executions.indices
205+
val clusterInfo = listOf(Pair(UtClusterInfo("FUZZER"), intRange))
203206
CgMethodTestSet(
204207
executableId = methodIds[testSet.method] as ExecutableId,
205-
executions = testSet.executions
208+
executions = testSet.executions,
209+
clustersInfo = clusterInfo,
206210
)
207211
},
208212
allImports
@@ -299,4 +303,4 @@ object PythonTestGenerationProcessor {
299303
} else {
300304
paths
301305
}
302-
}
306+
}

0 commit comments

Comments
 (0)