Fix the bug where SetProxy's __ior__, __ixor__, __iand__, and __isub__ return a set instead of a SetProxy

Author: mmingyu

Lib/multiprocessing/managers.py

@@ -1201,6 +1201,27 @@ def __ior__(self, value):
 del _set_proxy_methods
 
 class SetProxy(_BaseSetProxy):
+    def __ior__(self, value):
+        self._callmethod('__ior__', (value,))
+        return self
+    def __iand__(self, value):
+        self._callmethod('__iand__', (value,))
+        return self
+    def __ixor__(self, value):
+        self._callmethod('__ixor__', (value,))
+        return self
+    def __isub__(self, value):
+        self._callmethod('__isub__', (value,))
+        return self
+    def __le__(self, value):
+        return self._callmethod('__le__', (value,))
+    def __lt__(self, value):
+        return self._callmethod('__lt__', (value,))
+    def __ge__(self, value):
+        return self._callmethod('__ge__', (value,))
+    def __gt__(self, value):
+        return self._callmethod('__gt__', (value,))
+
     __class_getitem__ = classmethod(types.GenericAlias)
 
 collections.abc.MutableMapping.register(_BaseSetProxy)

Lib/test/_test_multiprocessing.py

@@ -6454,6 +6454,7 @@ def _test_set_operator_symbols(cls, obj):
         case.assertSetEqual(result, {'a', 'b', 'c', 'd', 'e'})
         obj |= {'d', 'e'}
         case.assertSetEqual(obj, {'a', 'b', 'c', 'd', 'e'})
+        case.assertIsInstance(obj, multiprocessing.managers.SetProxy)
         obj.clear()
         obj.update(['a', 'b', 'c'])
         result = {'a', 'b', 'd'} - obj
@@ -6462,6 +6463,7 @@ def _test_set_operator_symbols(cls, obj):
         case.assertSetEqual(result, {'c'})
         obj -= {'a', 'b'}
         case.assertSetEqual(obj, {'c'})
+        case.assertIsInstance(obj, multiprocessing.managers.SetProxy)
         obj.clear()
         obj.update(['a', 'b', 'c'])
         result = {'b', 'c', 'd'} ^ obj
@@ -6470,13 +6472,15 @@ def _test_set_operator_symbols(cls, obj):
         case.assertSetEqual(result, {'a', 'd'})
         obj ^= {'b', 'c', 'd'}
         case.assertSetEqual(obj, {'a', 'd'})
+        case.assertIsInstance(obj, multiprocessing.managers.SetProxy)
         obj.clear()
         obj.update(['a', 'b', 'c'])
         result = obj & {'b', 'c', 'd'}
         case.assertSetEqual(result, {'b', 'c'})
         result = {'b', 'c', 'd'} & obj
         case.assertSetEqual(result, {'b', 'c'})
         obj &= {'b', 'c', 'd'}
+        case.assertIsInstance(obj, multiprocessing.managers.SetProxy)
         case.assertSetEqual(obj, {'b', 'c'})
         obj.clear()
         obj.update(['a', 'b', 'c'])