Fix imports

Author: tamarinvs19

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/PythonCodeGenerator.kt

@@ -5,14 +5,18 @@ import org.utbot.framework.codegen.CodeGeneratorResult
 import org.utbot.framework.codegen.domain.ForceStaticMocking
 import org.utbot.framework.codegen.domain.HangingTestsTimeout
 import org.utbot.framework.codegen.domain.ParametrizedTestSource
-import org.utbot.framework.codegen.PythonImport
 import org.utbot.framework.codegen.domain.RuntimeExceptionTestsBehaviour
 import org.utbot.framework.codegen.domain.StaticsMocking
 import org.utbot.framework.codegen.domain.TestFramework
+import org.utbot.framework.codegen.domain.context.CgContext
+import org.utbot.framework.codegen.domain.models.CgLiteral
 import org.utbot.framework.codegen.domain.models.CgMethodTestSet
+import org.utbot.framework.codegen.domain.models.CgVariable
+import org.utbot.framework.codegen.domain.models.CgAssignment
 import org.utbot.framework.codegen.domain.models.TestClassModel
-import org.utbot.framework.codegen.domain.context.CgContext
 import org.utbot.framework.codegen.renderer.CgAbstractRenderer
+import org.utbot.framework.codegen.renderer.CgPrinterImpl
+import org.utbot.framework.codegen.renderer.CgRendererContext
 import org.utbot.python.PythonMethod
 import org.utbot.python.code.AnnotationProcessor.getModulesFromAnnotation
 import org.utbot.python.framework.api.python.NormalizedPythonAnnotation
@@ -23,6 +27,7 @@ import org.utbot.python.framework.api.python.util.pythonStrClassId
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.ExecutableId
 import org.utbot.framework.plugin.api.MockFramework
+import org.utbot.framework.plugin.api.UtModel
 import org.utbot.python.framework.codegen.PythonCgLanguageAssistant
 import org.utbot.python.framework.codegen.model.constructor.tree.PythonCgTestClassConstructor
 import org.utbot.python.framework.codegen.model.constructor.visitor.CgPythonRenderer

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/PythonImports.kt

@@ -1,7 +1,5 @@
 package org.utbot.python.framework.codegen.model
 
-import org.utbot.framework.codegen.Import
-
 import org.utbot.framework.codegen.domain.Import
 
 sealed class PythonImport(order: Int) : Import(order) {

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/constructor/util/ConstructorUtils.kt

@@ -2,11 +2,10 @@ package org.utbot.python.framework.codegen.model.constructor.util
 
 import kotlinx.collections.immutable.PersistentList
 import kotlinx.collections.immutable.PersistentSet
-import org.utbot.framework.codegen.PythonImport
-import org.utbot.framework.codegen.PythonSysPathImport
 import org.utbot.framework.codegen.domain.context.CgContextOwner
 import org.utbot.python.framework.api.python.PythonClassId
 import org.utbot.python.framework.api.python.PythonMethodId
+import org.utbot.python.framework.codegen.model.PythonUserImport
 
 internal fun CgContextOwner.importIfNeeded(method: PythonMethodId) {
     collectedImports += PythonUserImport(method.moduleName)