From 2c7716fed03049a72d68096908c334e5c9bb1680 Mon Sep 17 00:00:00 2001 From: Damien George Date: Wed, 28 Sep 2016 11:06:18 +1000 Subject: [PATCH] py/objset: Ensure that use of frozenset.update raises an exception. --- py/objset.c | 1 + tests/basics/frozenset_add.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/py/objset.c b/py/objset.c index 0425499008..6b6f95f9b2 100644 --- a/py/objset.c +++ b/py/objset.c @@ -435,6 +435,7 @@ STATIC void set_update_int(mp_obj_set_t *self, mp_obj_t other_in) { } STATIC mp_obj_t set_update(size_t n_args, const mp_obj_t *args) { + check_set(args[0]); for (mp_uint_t i = 1; i < n_args; i++) { set_update_int(MP_OBJ_TO_PTR(args[0]), args[i]); } diff --git a/tests/basics/frozenset_add.py b/tests/basics/frozenset_add.py index 50615775bd..415a8c2e13 100644 --- a/tests/basics/frozenset_add.py +++ b/tests/basics/frozenset_add.py @@ -10,3 +10,8 @@ try: print(s.add(5)) except AttributeError: print("AttributeError") + +try: + print(s.update([5])) +except AttributeError: + print("AttributeError")