diff --git a/py/objset.c b/py/objset.c index 8bd006a761..40f19c85d3 100644 --- a/py/objset.c +++ b/py/objset.c @@ -1,5 +1,6 @@ #include #include +#include #include #include "nlr.h" @@ -114,6 +115,20 @@ static mp_obj_t set_clear(mp_obj_t self_in) { } static MP_DEFINE_CONST_FUN_OBJ_1(set_clear_obj, set_clear); +static mp_obj_t set_copy(mp_obj_t self_in) { + assert(MP_OBJ_IS_TYPE(self_in, &set_type)); + mp_obj_set_t *self = self_in; + + mp_obj_set_t *other = m_new_obj(mp_obj_set_t); + other->base.type = &set_type; + mp_set_init(&other->set, self->set.alloc); + other->set.used = self->set.used; + memcpy(other->set.table, self->set.table, self->set.alloc * sizeof(mp_obj_t)); + + return other; +} +static MP_DEFINE_CONST_FUN_OBJ_1(set_copy_obj, set_copy); + /******************************************************************************/ /* set constructors & public C API */ @@ -122,6 +137,7 @@ static MP_DEFINE_CONST_FUN_OBJ_1(set_clear_obj, set_clear); static const mp_method_t set_type_methods[] = { { "add", &set_add_obj }, { "clear", &set_clear_obj }, + { "copy", &set_copy_obj }, { NULL, NULL }, // end-of-list sentinel }; diff --git a/tests/basics/tests/set_copy.py b/tests/basics/tests/set_copy.py new file mode 100644 index 0000000000..2ea308b0db --- /dev/null +++ b/tests/basics/tests/set_copy.py @@ -0,0 +1,8 @@ +s = {1, 2, 3, 4} +t = s.copy() +s.add(5) +t.add(7) +for i in s, t: + l = list(i) + l.sort() + print(l)