Merge pull request #3947 from jopemachine/edit-defaultdict

youknowone · web-flow · commit 722f8fdedd7f · 2022-07-22T18:17:27.000+09:00
Add `or`, `ror` and`ior` to `defaultdict`
diff --git a/Lib/collections/_defaultdict.py b/Lib/collections/_defaultdict.py
@@ -39,4 +39,20 @@ def __reduce__(self):
             args = ()
         return type(self), args, None, None, iter(self.items())
 
+    def __or__(self, other):
+        if not isinstance(other, dict):
+            return NotImplemented
+
+        new = defaultdict(self.default_factory, self)
+        new.update(other)
+        return new
+
+    def __ror__(self, other):
+        if not isinstance(other, dict):
+            return NotImplemented
+
+        new = defaultdict(self.default_factory, other)
+        new.update(self)
+        return new
+
 defaultdict.__module__ = 'collections'
diff --git a/Lib/test/test_defaultdict.py b/Lib/test/test_defaultdict.py
@@ -150,8 +150,6 @@ def test_pickling(self):
             o = pickle.loads(s)
             self.assertEqual(d, o)
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_union(self):
         i = defaultdict(int, {1: 1, 2: 2})
         s = defaultdict(str, {0: "zero", 1: "one"})
