[Andrew Polyakov] Modify NDArrayValueProvider

Author: AndrewPolyakov1

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

@@ -0,0 +1,40 @@
+package org.utbot.python.fuzzing.provider
+
+import mu.KotlinLogging
+import org.utbot.fuzzing.Routine
+import org.utbot.fuzzing.Seed
+import org.utbot.python.framework.api.python.PythonTree
+import org.utbot.python.framework.api.python.util.pythonNdarrayClassId
+import org.utbot.python.fuzzing.FuzzedUtType
+import org.utbot.python.fuzzing.FuzzedUtType.Companion.activateAnyIf
+import org.utbot.python.fuzzing.FuzzedUtType.Companion.toFuzzed
+import org.utbot.python.fuzzing.PythonFuzzedValue
+import org.utbot.python.fuzzing.PythonMethodDescription
+import org.utbot.python.fuzzing.PythonValueProvider
+import org.utpython.types.pythonAnnotationParameters
+
+
+private val logger = KotlinLogging.logger {}
+
+object NDArrayValueProvider : PythonValueProvider {
+    override fun accept(type: FuzzedUtType): Boolean {
+        return type.pythonTypeName() == pythonNdarrayClassId.canonicalName
+    }
+
+    override fun generate(
+        description: PythonMethodDescription, type: FuzzedUtType
+    ) = sequence {
+        val param = type.utType.pythonAnnotationParameters()
+        yield(Seed.Collection( //TODO: Rewrite to construct NDArray objects
+            construct = Routine.Collection {
+                PythonFuzzedValue(
+                    PythonTree.NdarrayNode(
+                        emptyMap<Int, PythonTree.PythonTreeNode>().toMutableMap(),
+                    ), "%var% = ${type.pythonTypeRepresentation()}"
+                )
+            },
+            modify = Routine.ForEach(param.toFuzzed().activateAnyIf(type)) { self, i, values ->
+                (self.tree as PythonTree.NdarrayNode).items[i] = values.first().tree
+            }))
+    }
+}