Forks in python fuzzing and refactoring

utbot-cli-python/src/main/kotlin/org/utbot/cli/language/python/PythonCliProcessor.kt

@@ -10,6 +10,7 @@ class PythonCliProcessor(
     private val output: String,
     private val logger: KLogger,
     private val coverageOutput: String?,
+    private val executionCounterOutput: String?,
 ) : PythonTestGenerationProcessor() {
 
     override fun saveTests(testsCode: String) {
@@ -22,9 +23,18 @@ class PythonCliProcessor(
         )
     }
 
+    private fun getExecutionsNumber(testSets: List<PythonTestSet>): Int {
+        return testSets.sumOf { it.executionsNumber }
+    }
+
     override fun processCoverageInfo(testSets: List<PythonTestSet>) {
-        val coverageReport = getStringCoverageInfo(testSets)
-        val output = coverageOutput ?: return
-        writeToFileAndSave(output, coverageReport)
+        coverageOutput?.let { output ->
+            val coverageReport = getStringCoverageInfo(testSets)
+            writeToFileAndSave(output, coverageReport)
+        }
+        executionCounterOutput?.let { executionOutput ->
+            val executionReport = "{\"executions\": ${getExecutionsNumber(testSets)}}"
+            writeToFileAndSave(executionOutput, executionReport)
+        }
     }
 }

utbot-cli-python/src/main/kotlin/org/utbot/cli/language/python/PythonGenerateTestsCommand.kt

@@ -76,6 +76,11 @@ class PythonGenerateTestsCommand : CliktCommand(
         help = "File to write coverage report."
     )
 
+    private val executionCounterOutput by option(
+        "--executions-counter",
+        help = "File to write number of executions."
+    )
+
     private val installRequirementsIfMissing by option(
         "--install-requirements",
         help = "Install Python requirements if missing."
@@ -255,6 +260,7 @@ class PythonGenerateTestsCommand : CliktCommand(
             output.toAbsolutePath(),
             logger,
             coverageOutput?.toAbsolutePath(),
+            executionCounterOutput?.toAbsolutePath(),
         )
 
         logger.info("Loading information about Python types...")
@@ -270,7 +276,8 @@ class PythonGenerateTestsCommand : CliktCommand(
                     testSet.executions.filterNot { it.result is UtExecutionSuccess },
                     testSet.errors,
                     testSet.mypyReport,
-                    testSet.classId
+                    testSet.classId,
+                    testSet.executionsNumber
                 )
             }
         }

utbot-python-executor/src/main/python/utbot_executor/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utbot-executor"
-version = "1.4.44"
+version = "1.7.0"
 description = ""
 authors = ["Vyacheslav Tamarin <vyacheslav.tamarin@yandex.ru>"]
 readme = "README.md"

utbot-python-executor/src/main/resources/utbot_executor_version

@@ -1 +1 @@
-1.4.44
+1.7.0