Merge pull request #458 from v923z/sum-fix

Sum fix

Author: v923z

code/numpy/numerical.c

@@ -293,7 +293,7 @@ static mp_obj_t numerical_sum_mean_std_ndarray(ndarray_obj_t *ndarray, mp_obj_t
             if(ndarray->dtype == NDARRAY_FLOAT) {
                 return mp_obj_new_float(M * ndarray->len);
             } else {
-                return mp_obj_new_int((int32_t)(M * ndarray->len));
+                return mp_obj_new_int((int32_t)MICROPY_FLOAT_C_FUN(round)(M * ndarray->len));
             }
         } else if(optype == NUMERICAL_MEAN) {
             return mp_obj_new_float(M);

code/ulab.c

@@ -33,7 +33,7 @@
 #include "user/user.h"
 #include "utils/utils.h"
 
-#define ULAB_VERSION 3.3.6
+#define ULAB_VERSION 3.3.7
 #define xstr(s) str(s)
 #define str(s) #s
 #define ULAB_VERSION_STRING xstr(ULAB_VERSION) xstr(-) xstr(ULAB_MAX_DIMS) xstr(D)

docs/ulab-change-log.md

@@ -1,3 +1,9 @@
+Tue, 30 Nov 2021
+
+version 3.3.7
+
+    fix sum() for integer/Boolean types
+
 Sat, 20 Nov 2021
 
 version 3.3.6

tests/1d/numpy/sum.py

@@ -0,0 +1,21 @@
+from ulab import numpy as np
+
+r = range(15)
+
+a = np.array(r, dtype=np.uint8)
+print(np.sum(a))
+
+a = np.array(r, dtype=np.int8)
+print(np.sum(a))
+
+a = np.array(r, dtype=np.uint16)
+print(np.sum(a))
+
+a = np.array(r, dtype=np.int16)
+print(np.sum(a))
+
+a = np.array(r, dtype=np.float)
+print(np.sum(a))
+
+a = np.array([False] + [True]*15, dtype=np.bool)
+print(np.sum(a))

tests/1d/numpy/sum.py.exp

@@ -0,0 +1,6 @@
+105
+105
+105
+105
+105.0
+15