mirror
/
micropython
kopia lustrzana https://github.com/micropython/micropython
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

py/objset: Include the failed key in a KeyError raised from set.remove.

pull/3310/head
Damien George 2017-10-03 18:03:06 +11:00
rodzic 2ac1364688
commit 1394258f37
2 zmienionych plików z 3 dodań i 3 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

2
py/objset.c
Wyświetl plik

@ -380,7 +380,7 @@ STATIC mp_obj_t set_remove(mp_obj_t self_in, mp_obj_t item) {
check_set(self_in); check_set(self_in);
mp_obj_set_t *self = MP_OBJ_TO_PTR(self_in); mp_obj_set_t *self = MP_OBJ_TO_PTR(self_in);
if (mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_REMOVE_IF_FOUND) == MP_OBJ_NULL) { if (mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_REMOVE_IF_FOUND) == MP_OBJ_NULL) {
nlr_raise(mp_obj_new_exception(&mp_type_KeyError)); nlr_raise(mp_obj_new_exception_arg1(&mp_type_KeyError, item));
} }
return mp_const_none; return mp_const_none;
} }

4
tests/basics/set_remove.py
Wyświetl plik

@ -4,8 +4,8 @@ print(s.remove(1))
print(list(s)) print(list(s))
try: try:
print(s.remove(1), "!!!") print(s.remove(1), "!!!")
except KeyError: except KeyError as er:
pass print('KeyError', er.args[0])
else: else:
print("failed to raise KeyError") print("failed to raise KeyError")