Try to add complex and refactoring

Author: tamarinvs19

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

@@ -222,10 +222,10 @@ class PythonEngine(
             if (coverageLimit < 0)
                 return@PythonFuzzing PythonFeedback(control = Control.STOP)
 
-            if (parameterValues.all { it.classId.name == "UNDEF_VALUE" }) {
-                coverageLimit--
-                return@PythonFuzzing PythonFeedback(control = Control.PASS)
-            }
+//            if (parameterValues.all { it.classId.name == "UNDEF_VALUE" }) {
+//                coverageLimit--
+//                return@PythonFuzzing PythonFeedback(control = Control.PASS)
+//            }
 
             val (thisObject, modelList) =
                 if (methodUnderTest.containingPythonClassId == null)

utbot-python/src/main/kotlin/org/utbot/python/framework/api/python/util/PythonIdUtils.kt

@@ -10,9 +10,11 @@ import org.utbot.python.framework.api.python.PythonSetModel
 
 // none annotation can be used in code only since Python 3.10
 val pythonNoneClassId = PythonClassId("types.NoneType")
+val pythonObjectClassId = PythonClassId("builtins.object")
 val pythonAnyClassId = NormalizedPythonAnnotation("typing.Any")
 val pythonIntClassId = PythonClassId("builtins.int")
 val pythonFloatClassId = PythonClassId("builtins.float")
+val pythonComplexClassId = PythonClassId("builtins.complex")
 val pythonStrClassId = PythonClassId("builtins.str")
 val pythonBoolClassId = PythonBoolModel.classId
 val pythonRangeClassId = PythonClassId("builtins.range")

utbot-python/src/main/kotlin/org/utbot/python/fuzzing/PythonApi.kt

@@ -12,7 +12,9 @@ import org.utbot.python.framework.api.python.PythonClassId
 import org.utbot.python.framework.api.python.PythonTree
 import org.utbot.python.framework.api.python.PythonTreeModel
 import org.utbot.python.framework.api.python.util.pythonNoneClassId
+import org.utbot.python.framework.api.python.util.pythonObjectClassId
 import org.utbot.python.fuzzing.provider.BoolValueProvider
+import org.utbot.python.fuzzing.provider.ComplexValueProvider
 import org.utbot.python.fuzzing.provider.DictValueProvider
 import org.utbot.python.fuzzing.provider.FloatValueProvider
 import org.utbot.python.fuzzing.provider.IntValueProvider
@@ -24,7 +26,7 @@ import org.utbot.python.fuzzing.provider.StrValueProvider
 import org.utbot.python.fuzzing.provider.TupleFixSizeValueProvider
 import org.utbot.python.fuzzing.provider.TupleValueProvider
 import org.utbot.python.fuzzing.provider.UnionValueProvider
-import org.utbot.python.fuzzing.value.UndefValue
+import org.utbot.python.fuzzing.value.ObjectValue
 import org.utbot.python.newtyping.PythonProtocolDescription
 import org.utbot.python.newtyping.PythonSubtypeChecker
 import org.utbot.python.newtyping.PythonTypeStorage
@@ -60,6 +62,7 @@ fun pythonDefaultValueProviders(idGenerator: IdGenerator<Long>) = listOf(
     BoolValueProvider,
     IntValueProvider,
     FloatValueProvider,
+    ComplexValueProvider,
     StrValueProvider,
     ListValueProvider,
     SetValueProvider,
@@ -81,6 +84,7 @@ class PythonFuzzing(
                 providers += provider.generate(description, type)
             }
         }
+        logger.info("Default ${type.meta} providers: $providers")
         return providers
     }
 
@@ -92,9 +96,9 @@ class PythonFuzzing(
             }
             subtypes.forEach {
                 providers += generateDefault(description, it, idGenerator)
-//                providers += generateSubtype(description, it, idGenerator)
             }
         }
+        logger.info("Subtype ${type.meta} providers: $providers")
         return providers
     }
 
@@ -106,7 +110,8 @@ class PythonFuzzing(
         providers += generateSubtype(description, type, idGenerator)
 
         if (providers.toList().isEmpty()) {
-            providers += Seed.Known(UndefValue()) {PythonTreeModel(PythonTree.fromNone(), PythonClassId("UNDEF_VALUE"))}
+            logger.info("Add object provider for ${type.meta}")
+            providers += Seed.Known(ObjectValue()) {PythonTreeModel(PythonTree.fromNone(), pythonObjectClassId)}
         }
 
         return providers

utbot-python/src/main/kotlin/org/utbot/python/fuzzing/provider/ComplexValueProvider.kt

@@ -0,0 +1,43 @@
+package org.utbot.python.fuzzing.provider
+
+import org.utbot.fuzzing.Routine
+import org.utbot.fuzzing.Seed
+import org.utbot.fuzzing.ValueProvider
+import org.utbot.fuzzing.seeds.BitVectorValue
+import org.utbot.fuzzing.seeds.Signed
+import org.utbot.python.framework.api.python.PythonClassId
+import org.utbot.python.framework.api.python.PythonTree
+import org.utbot.python.framework.api.python.PythonTreeModel
+import org.utbot.python.framework.api.python.util.pythonComplexClassId
+import org.utbot.python.framework.api.python.util.pythonFloatClassId
+import org.utbot.python.framework.api.python.util.pythonIntClassId
+import org.utbot.python.fuzzing.PythonMethodDescription
+import org.utbot.python.fuzzing.provider.utils.isAny
+import org.utbot.python.newtyping.PythonConcreteCompositeTypeDescription
+import org.utbot.python.newtyping.general.Type
+
+object ComplexValueProvider : ValueProvider<Type, PythonTreeModel, PythonMethodDescription> {
+    override fun accept(type: Type): Boolean {
+        val meta = type.meta
+        if (meta is PythonConcreteCompositeTypeDescription) {
+            return meta.name.toString() == "builtins.complex"
+        }
+        return type.isAny()
+    }
+
+    override fun generate(description: PythonMethodDescription, type: Type) = sequence {
+        val meta = type.meta as PythonConcreteCompositeTypeDescription
+        yield(Seed.Recursive(
+            construct = Routine.Create(emptyList<Type>()) { v ->
+                PythonTreeModel(
+                    PythonTree.PrimitiveNode(
+                        pythonComplexClassId,
+                        "complex('${v[0]}+${v[1]}j')"
+                    ),
+                    pythonComplexClassId,
+                )
+            },
+            empty = Routine.Empty { PythonTreeModel(PythonTree.fromObject(), PythonClassId(meta.name.toString())) }
+        ))
+    }
+}

utbot-python/src/main/kotlin/org/utbot/python/fuzzing/provider/UnionValueProvider.kt

@@ -31,7 +31,7 @@ object UnionValueProvider : ValueProvider<Type, PythonTreeModel, PythonMethodDes
         params.forEach { unionParam ->
             yield(Seed.Recursive(
                 construct = Routine.Create(listOf(unionParam)) { v -> v.first() },
-                empty = Routine.Empty { PythonTreeModel(PythonTree.fromNone(), PythonClassId(meta.name.toString())) }
+                empty = Routine.Empty { PythonTreeModel(PythonTree.fromObject(), PythonClassId(meta.name.toString())) }
             ))
         }
     }

utbot-python/src/main/kotlin/org/utbot/python/fuzzing/value/UndefValue.kt

@@ -3,7 +3,7 @@ package org.utbot.python.fuzzing.value
 import org.utbot.fuzzing.Mutation
 import org.utbot.fuzzing.seeds.KnownValue
 
-class UndefValue : KnownValue {
+class ObjectValue : KnownValue {
     override fun mutations(): List<Mutation<KnownValue>> {
         return emptyList()
     }