Fix FakeNode in ComplexValueProvider

tamarinvs19 · tamarinvs19 · commit 6cd18c817814 · 2023-11-21T17:46:51.000+03:00
diff --git a/utbot-python/src/main/kotlin/org/utbot/python/fuzzing/provider/ComplexValueProvider.kt b/utbot-python/src/main/kotlin/org/utbot/python/fuzzing/provider/ComplexValueProvider.kt
@@ -24,33 +24,37 @@ object ComplexValueProvider : ValueProvider<UtType, PythonFuzzedValue, PythonMet
                 description.pythonTypeStorage.pythonInt
             )
         )
+        val emptyValue =
+            PythonFuzzedValue(
+                PythonTree.PrimitiveNode(
+                    pythonComplexClassId,
+                    "complex()"
+                ),
+                "%var% = ${type.pythonTypeRepresentation()}"
+            )
         yield(Seed.Recursive(
             construct = Routine.Create(
                 listOf(
                     numberType,
                     numberType
                 )
             ) { v ->
-                val real = v[0].tree as PythonTree.PrimitiveNode
-                val imag = v[1].tree as PythonTree.PrimitiveNode
-                val repr = "complex(real=${real.repr}, imag=${imag.repr})"
-                PythonFuzzedValue(
-                    PythonTree.PrimitiveNode(
-                        pythonComplexClassId,
-                        repr
-                    ),
-                    "%var% = $repr"
-                )
+                if (v[0].tree is PythonTree.FakeNode || v[1].tree is PythonTree.FakeNode) {
+                    emptyValue
+                } else {
+                    val real = v[0].tree as PythonTree.PrimitiveNode
+                    val imag = v[1].tree as PythonTree.PrimitiveNode
+                    val repr = "complex(real=${real.repr}, imag=${imag.repr})"
+                    PythonFuzzedValue(
+                        PythonTree.PrimitiveNode(
+                            pythonComplexClassId,
+                            repr
+                        ),
+                        "%var% = $repr"
+                    )
+                }
             },
-            empty = Routine.Empty {
-                PythonFuzzedValue(
-                    PythonTree.PrimitiveNode(
-                        pythonComplexClassId,
-                        "complex()"
-                    ),
-                    "%var% = ${type.pythonTypeRepresentation()}"
-                )
-            }
+            empty = Routine.Empty { emptyValue }
         ))
     }
 }
