diff --git a/py/modstruct.c b/py/modstruct.c index 0c4af707d6..1ba974d55d 100644 --- a/py/modstruct.c +++ b/py/modstruct.c @@ -28,6 +28,7 @@ #include #include +#include "py/runtime.h" #include "py/builtin.h" #include "py/objtuple.h" #include "py/binary.h" @@ -104,7 +105,9 @@ STATIC mp_obj_t struct_calcsize(mp_obj_t fmt_in) { } if (cnt > 1) { // TODO: count spec support only for string len - assert(*fmt == 's'); + if (*fmt != 's') { + mp_not_implemented("count>1"); + } } mp_uint_t sz; @@ -140,7 +143,9 @@ STATIC mp_obj_t struct_unpack(mp_obj_t fmt_in, mp_obj_t data_in) { } if (sz > 1) { // TODO: size spec support only for string len - assert(*fmt == 's'); + if (*fmt != 's') { + mp_not_implemented("count>1"); + } } mp_obj_t item; if (*fmt == 's') { @@ -173,7 +178,9 @@ STATIC mp_obj_t struct_pack(mp_uint_t n_args, const mp_obj_t *args) { } if (sz > 1) { // TODO: size spec support only for string len - assert(*fmt == 's'); + if (*fmt != 's') { + mp_not_implemented("count>1"); + } } if (*fmt == 's') {