[Andrew Polyakov] Change max shape value to reduce memory consumption

Author: AndrewPolyakov1

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

@@ -15,10 +15,13 @@ import kotlin.math.abs
 
 
 private val logger = KotlinLogging.logger {}
+private const val SHAPE_SIZE = 3
+private const val MAX_SHAPE_DIGITS = 2
 
 class NDArrayValueProvider(
     private val typeStorage: PythonTypeHintsStorage
 ) : PythonValueProvider {
+
     override fun accept(type: FuzzedUtType): Boolean {
         return type.pythonTypeName() == pythonNdarrayClassId.canonicalName
     }
@@ -41,20 +44,25 @@ class NDArrayValueProvider(
             ) {
                 PythonFuzzedValue(
                     PythonTree.NDArrayNode(
-                        emptyMap<Int, PythonTree.PythonTreeNode>().toMutableMap(), // Generate new Python IntNode
+                        emptyMap<Int, PythonTree.PythonTreeNode>().toMutableMap(),
                         ((it.first().tree as PythonTree.ListNode).items as Map<Int, PythonTree.PrimitiveNode>).values.map { node ->
-                            abs(node.repr.take(2).toInt())
-                        }
+                            abs(node.repr.take(MAX_SHAPE_DIGITS).toInt()) % 10
+                        }.take(SHAPE_SIZE).let { self ->
+                            if (self.fold(1, Int::times) == 0){
+                                listOf(0)
+                            } else {
+                                self
+                            }
+                        } // TODO: Rethink logic
                     ), "%var% = ${type.pythonTypeRepresentation()}"
                 )
             },
             modify = sequence {
-                yield(Routine.Call((0 until 10000).map { param[1] }.toFuzzed()) { instance, arguments ->
+                yield(Routine.Call((0 until 1000000).map { param[1] }.toFuzzed()) { instance, arguments ->
                     val obj = instance.tree as PythonTree.NDArrayNode
                     (0 until obj.dimensions.fold(1, Int::times)).map {
-                        obj.items[it] = arguments.get(it).tree
+                        obj.items[it] = arguments[it].tree
                     }
-
                 })
             },
             empty = Routine.Empty { PythonFuzzedValue(PythonTree.FakeNode) }