Implemented set.isdisjoint

1 files changed, 16 insertions, 0 deletions

diff --git a/py/objset.c b/py/objset.c
index 144bb6daa8..afc426b3b0 100644
--- a/py/objset.c
+++ b/py/objset.c
@@ -212,6 +212,21 @@ static mp_obj_t set_intersect_update(mp_obj_t self_in, mp_obj_t other) {
 }
 static MP_DEFINE_CONST_FUN_OBJ_2(set_intersect_update_obj, set_intersect_update);
 
+static mp_obj_t set_isdisjoint(mp_obj_t self_in, mp_obj_t other) {
+    assert(MP_OBJ_IS_TYPE(self_in, &set_type));
+    mp_obj_set_t *self = self_in;
+
+    mp_obj_t iter = rt_getiter(other);
+    mp_obj_t next;
+    while ((next = rt_iternext(iter)) != mp_const_stop_iteration) {
+        if (mp_set_lookup(&self->set, next, MP_MAP_LOOKUP)) {
+            return mp_const_false;
+        }
+    }
+    return mp_const_true;
+}
+static MP_DEFINE_CONST_FUN_OBJ_2(set_isdisjoint_obj, set_isdisjoint);
+
 
 /******************************************************************************/
 /* set constructors & public C API                                            */
@@ -226,6 +241,7 @@ static const mp_method_t set_type_methods[] = {
     { "difference_update", &set_diff_update_obj },
     { "intersection", &set_intersect_obj },
     { "intersection_update", &set_intersect_update_obj },
+    { "isdisjoint", &set_isdisjoint_obj },
     { NULL, NULL }, // end-of-list sentinel
 };