From e9739e33156921ec111d60c6a84a1355a195a875 Mon Sep 17 00:00:00 2001 From: Paul Sokolovsky Date: Sat, 18 Jun 2016 01:30:49 +0300 Subject: [PATCH] extmod/modbtree: __getitem__() should raise KeyError for non-existing key. --- extmod/modbtree.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extmod/modbtree.c b/extmod/modbtree.c index eff2cd1f5a..345b4d1618 100644 --- a/extmod/modbtree.c +++ b/extmod/modbtree.c @@ -203,7 +203,7 @@ STATIC mp_obj_t btree_subscr(mp_obj_t self_in, mp_obj_t index, mp_obj_t value) { key.data = (void*)mp_obj_str_get_data(index, &key.size); int res = __bt_get(self->db, &key, &val, 0); if (res == RET_SPECIAL) { - return mp_const_none; + nlr_raise(mp_obj_new_exception(&mp_type_KeyError)); } CHECK_ERROR(res); return mp_obj_new_bytes(val.data, val.size);