Added division of timeout for several function (#1847)

Added division of timeout

Author: tochilinak

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

@@ -39,7 +39,6 @@ class PythonTestCaseGenerator(
     private val fileOfMethod: String,
     private val isCancelled: () -> Boolean,
     private val timeoutForRun: Long = 0,
-    private val until: Long = 0,
     private val sourceFileContent: String,
     private val mypyStorage: MypyAnnotationStorage,
     private val mypyReportLine: List<MypyReportLine>,
@@ -145,7 +144,7 @@ class PythonTestCaseGenerator(
         }.take(maxSubstitutions)
     }
 
-    fun generate(methodDescription: PythonMethodHeader): PythonTestSet {
+    fun generate(methodDescription: PythonMethodHeader, until: Long): PythonTestSet {
         storageForMypyMessages.clear()
 
         val typeStorage = PythonTypeStorage.get(mypyStorage)

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

@@ -98,15 +98,17 @@ object PythonTestGenerationProcessor {
                 fileOfMethod = pythonFilePath,
                 isCancelled = isCanceled,
                 timeoutForRun = timeoutForRun,
-                until = startTime + timeout,
                 sourceFileContent = pythonFileContent,
                 mypyStorage = mypyStorage,
                 mypyReportLine = report,
                 mypyConfigFile = mypyConfigFile,
             )
 
-            val tests = pythonMethods.map { method ->
-                testCaseGenerator.generate(method)
+            val until = startTime + timeout
+            val tests = pythonMethods.mapIndexed { index, method ->
+                val methodsLeft = pythonMethods.size - index
+                val localUntil = (until - System.currentTimeMillis()) / methodsLeft + System.currentTimeMillis()
+                testCaseGenerator.generate(method, localUntil)
             }
 
             val (notEmptyTests, emptyTestSets) = tests.partition { it.executions.isNotEmpty() }