[Andrew Polyakov] Change codegen for type in ndarray

Author: AndrewPolyakov1

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/constructor/tree/PythonCgVariableConstructor.kt

@@ -79,8 +79,8 @@ class PythonCgVariableConstructor(cgContext: CgContext) : CgVariableConstructor(
             is PythonTree.NDArrayNode -> {
                 val items = objectNode.items.values.map { pythonBuildObject(it) }
                 val shape = objectNode.dimensions
-                val type = objectNode.items.values.firstOrNull()?.type?.typeName
-                Pair(CgPythonNdarray(items.map { it.first }, shape, type?: "int"), items.flatMap { it.second })
+                val type = objectNode.items.values.firstOrNull()?.type
+                Pair(CgPythonNdarray(items.map { it.first }, shape, type?:PythonClassId("int")), items.flatMap { it.second })
             }
 
             is PythonTree.TupleNode -> {

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/constructor/visitor/CgPythonRenderer.kt

@@ -563,7 +563,9 @@ internal class CgPythonRenderer(
     override fun visit(element: CgPythonNdarray) {
 
         print("numpy.ndarray(")
-        print("dtype=" + element.dtype + ",")
+        print("dtype=")
+        print(element.dtype.toString().removePrefix("builtins."))
+        print(",")
         print("shape=(")
 
         print(element.shape.joinToString(",") { it.toString() })

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

@@ -126,7 +126,7 @@ class CgPythonSet(
 class CgPythonNdarray(
     val elements: List<CgValue>,
     val shape: List<Int>,
-    val dtype: String
+    val dtype: PythonClassId
 ) : CgValue, CgPythonElement {
     override val type: PythonClassId = pythonNdarrayClassId
 }