Update ndarray.c

mdaeron · web-flow · commit af5d24f53130 · 2021-01-11T20:13:10.000+01:00
Fix bug in ndarray_tobytes()
diff --git a/code/ndarray.c b/code/ndarray.c
@@ -1554,7 +1554,7 @@ mp_obj_t ndarray_tobytes(mp_obj_t self_in) {
     if(!ndarray_is_dense(self)) {
         mp_raise_ValueError(translate("tobytes can be invoked for dense arrays only"));
     }
-    return mp_obj_new_bytearray_by_ref(self->len, self->array);
+    return mp_obj_new_bytearray_by_ref(self->itemsize * self->len, self->array);
 }
 
 MP_DEFINE_CONST_FUN_OBJ_1(ndarray_tobytes_obj, ndarray_tobytes);

Original file line number	Diff line number	Diff line change
`@@ -1554,7 +1554,7 @@ mp_obj_t ndarray_tobytes(mp_obj_t self_in) {`
`1554`	`1554`	`if(!ndarray_is_dense(self)) {`
`1555`	`1555`	`mp_raise_ValueError(translate("tobytes can be invoked for dense arrays only"));`
`1556`	`1556`	`}`
`1557`		`- return mp_obj_new_bytearray_by_ref(self->len, self->array);`
	`1557`	`+ return mp_obj_new_bytearray_by_ref(self->itemsize * self->len, self->array);`
`1558`	`1558`	`}`
`1559`	`1559`
`1560`	`1560`	`MP_DEFINE_CONST_FUN_OBJ_1(ndarray_tobytes_obj, ndarray_tobytes);`