Add copy method to list object

janczer · janczer · commit f45f307a555c · 2019-02-15T19:48:48.000+01:00
diff --git a/tests/snippets/set.py b/tests/snippets/set.py
@@ -83,3 +83,10 @@ def __hash__(self):
 else:
 	assert False, "TypeError was not raised"
 
+a = set([1,2,3])
+b = a.copy()
+assert len(a) == 3
+assert len(b) == 3
+b.clear()
+assert len(a) == 3
+assert len(b) == 0
diff --git a/vm/src/obj/objset.rs b/vm/src/obj/objset.rs
@@ -152,6 +152,13 @@ fn set_len(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     Ok(vm.context().new_int(elements.len()))
 }
 
+fn set_copy(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    trace!("set.copy called with: {:?}", args);
+    arg_check!(vm, args, required = [(s, Some(vm.ctx.set_type()))]);
+    let elements = get_elements(s);
+    Ok(PyObject::new(PyObjectPayload::Set { elements }, vm.ctx.set_type()))
+}
+
 fn set_repr(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(vm, args, required = [(o, Some(vm.ctx.set_type()))]);
 
@@ -450,6 +457,7 @@ pub fn init(context: &PyContext) {
     context.set_attr(&set_type, "add", context.new_rustfunc(set_add));
     context.set_attr(&set_type, "remove", context.new_rustfunc(set_remove));
     context.set_attr(&set_type, "clear", context.new_rustfunc(set_clear));
+    context.set_attr(&set_type, "copy", context.new_rustfunc(set_copy));
 
     let frozenset_type = &context.frozenset_type;
 
