Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-12-13 13:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

support del slice for list

Browse Source
This commit is contained in:
blueloveTH 2025-12-09 13:43:33 +08:00
parent f95b765207
commit 5c7fb79a14
2 changed files with 29 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/public/PyList.c
View File

@ -167,8 +167,26 @@ static bool list__setitem__(int argc, py_Ref argv) {
static bool list__delitem__(int argc, py_Ref argv) {
PY_CHECK_ARGC(2);
PY_CHECK_ARG_TYPE(1, tp_int);
List* self = py_touserdata(py_arg(0));
if(py_istype(py_arg(1), tp_slice)) {
int start, stop, step;
bool ok = pk__parse_int_slice(py_arg(1), self->length, &start, &stop, &step);
if(!ok) return false;
if(step != 1) return ValueError("slice step must be 1 for deletion");
int n = stop - start;
if(n > 0) {
py_TValue* p = self->data;
for(int i = stop; i < self->length; i++) {
p[start + i - stop] = p[i];
}
self->length -= n;
}
py_newnone(py_retval());
return true;
}
PY_CHECK_ARG_TYPE(1, tp_int);
int index = py_toint(py_arg(1));
if(!pk__normalize_index(&index, self->length)) return false;
c11_vector__erase(py_TValue, self, index);

10
tests/050_list.py
View File

@ -160,6 +160,16 @@ b = [(1, 2), (3, 3), (5, 1)]
b.sort(key=lambda x:x[1])
assert b == [(5, 1), (1, 2), (3,3)]
# test del slice
a = [1, 2, 3, 4]
b = a.copy(); del b[:2]; assert b == [3, 4]
b = a.copy(); del b[1:3]; assert b == [1, 4]
b = a.copy(); del b[2:]; assert b == [1, 2]
b = a.copy(); del b[:-1]; assert b == [4]
b = a.copy(); del b[-1:]; assert b == [1, 2, 3]
b = a.copy(); del b[:]; assert b == []
assert a == [1, 2, 3, 4]
# test cyclic reference
# a = []
# a.append(0)