Skip to content

Commit 694cbda

Browse files
committed
(frozen)set type: Added __doc__
1 parent 8f2c959 commit 694cbda

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

vm/src/obj/objset.rs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,11 @@ fn frozenset_repr(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
152152

153153
pub fn init(context: &PyContext) {
154154
let set_type = &context.set_type;
155+
156+
let set_doc = "set() -> new empty set object\n\
157+
set(iterable) -> new set object\n\n\
158+
Build an unordered collection of unique elements.";
159+
155160
context.set_attr(
156161
&set_type,
157162
"__contains__",
@@ -160,15 +165,26 @@ pub fn init(context: &PyContext) {
160165
context.set_attr(&set_type, "__len__", context.new_rustfunc(set_len));
161166
context.set_attr(&set_type, "__new__", context.new_rustfunc(set_new));
162167
context.set_attr(&set_type, "__repr__", context.new_rustfunc(set_repr));
168+
context.set_attr(&set_type, "__doc__", context.new_str(set_doc.to_string()));
163169
context.set_attr(&set_type, "add", context.new_rustfunc(set_add));
164170

165171
let frozenset_type = &context.frozenset_type;
172+
173+
let frozenset_doc = "frozenset() -> empty frozenset object\n\
174+
frozenset(iterable) -> frozenset object\n\n\
175+
Build an immutable unordered collection of unique elements.";
176+
166177
context.set_attr(
167178
&frozenset_type,
168179
"__contains__",
169180
context.new_rustfunc(set_contains),
170181
);
171182
context.set_attr(&frozenset_type, "__len__", context.new_rustfunc(set_len));
183+
context.set_attr(
184+
&frozenset_type,
185+
"__doc__",
186+
context.new_str(frozenset_doc.to_string()),
187+
);
172188
context.set_attr(
173189
&frozenset_type,
174190
"__repr__",

0 commit comments

Comments
 (0)