mirrors/micropython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bytearray: Support bytearray(int) constructor.

Browse Source 
To create bytearray of given length.
This commit is contained in:
Paul Sokolovsky 2014-04-08 04:42:44 +03:00
parent 72cfc6ef0a
commit b9cf3d3730
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
py/objarray.c
View File

@ -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);

1
tests/basics/bytearray1.py
View File

@ -1,3 +1,4 @@
print(bytearray(4))
a = bytearray([1, 2, 200])
print(a[0], a[2])
print(a[-1])