Fix set intersection_update implementation

crazymerlyn · crazymerlyn · commit b36c95b91eec · 2024-09-19T20:59:12.000+01:00
diff --git a/Lib/test/test_weakset.py b/Lib/test/test_weakset.py
@@ -322,8 +322,6 @@ def test_ixor(self):
             else:
                 self.assertNotIn(c, self.s)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_inplace_on_self(self):
         t = self.s.copy()
         t |= t
diff --git a/vm/src/builtins/set.rs b/vm/src/builtins/set.rs
@@ -391,16 +391,10 @@ impl PySetInner {
         others: impl std::iter::Iterator<Item = ArgIterable>,
         vm: &VirtualMachine,
     ) -> PyResult<()> {
-        let mut temp_inner = self.copy();
+        let temp_inner = self.fold_op(others, PySetInner::intersection, vm)?;
         self.clear();
-        for iterable in others {
-            for item in iterable.iter(vm)? {
-                let obj = item?;
-                if temp_inner.contains(&obj, vm)? {
-                    self.add(obj, vm)?;
-                }
-            }
-            temp_inner = self.copy()
+        for obj in temp_inner.elements() {
+            self.add(obj, vm)?;
         }
         Ok(())
     }
