py/objset: Ensure that use of frozenset.update raises an exception.

dpgeorge · dpgeorge · commit 2c7716fed030 · 2016-09-28T11:06:18.000+10:00
diff --git 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
@@ -10,3 +10,8 @@
     print(s.add(5))
 except AttributeError:
     print("AttributeError")
+
+try:
+    print(s.update([5]))
+except AttributeError:
+    print("AttributeError")

Original file line number	Diff line number	Diff line change
`@@ -435,6 +435,7 @@ STATIC void set_update_int(mp_obj_set_t *self, mp_obj_t other_in) {`
`435`	`435`	`}`
`436`	`436`
`437`	`437`	`STATIC mp_obj_t set_update(size_t n_args, const mp_obj_t *args) {`
	`438`	`+ check_set(args[0]);`
`438`	`439`	`for (mp_uint_t i = 1; i < n_args; i++) {`
`439`	`440`	`set_update_int(MP_OBJ_TO_PTR(args[0]), args[i]);`
`440`	`441`	`}`