bytearray: Support bytearray(int) constructor.
To create bytearray of given length.
diff --git a/py/objarray.c b/py/objarray.c
index 0383cb1..4f9fa49 100644
--- a/py/objarray.c
+++ b/py/objarray.c
@@ -91,6 +91,13 @@
// 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);