summaryrefslogtreecommitdiffstatshomepage
path: root/py/objarray.c
diff options
context:
space:
mode:
Diffstat (limited to 'py/objarray.c')
-rw-r--r--py/objarray.c7
1 files changed, 7 insertions, 0 deletions
diff --git a/py/objarray.c b/py/objarray.c
index 0383cb17b3..4f9fa49bcb 100644
--- a/py/objarray.c
+++ b/py/objarray.c
@@ -91,6 +91,13 @@ STATIC mp_obj_t array_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const m
// This is top-level factory function, not virtual method
// TODO: "bytearray" really should be type, not function
STATIC mp_obj_t mp_builtin_bytearray(mp_obj_t arg) {
+ if (MP_OBJ_IS_SMALL_INT(arg)) {
+ uint len = MP_OBJ_SMALL_INT_VALUE(arg);
+ mp_obj_array_t *o = array_new(BYTEARRAY_TYPECODE, len);
+ memset(o->items, 0, len);
+ return o;
+ }
+
return array_construct(BYTEARRAY_TYPECODE, arg);
}
MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_bytearray_obj, mp_builtin_bytearray);