-

cool-RR · cool-RR · commit 8f15e2810fad · 2014-10-08T22:58:26.000+03:00
diff --git a/source_py3/python_toolbox/nifty_collections/bagging.py b/source_py3/python_toolbox/nifty_collections/bagging.py
@@ -7,6 +7,7 @@
 import numbers
 import collections
 import functools
+import copy
 
 from python_toolbox import math_tools
 
@@ -334,6 +335,11 @@ def __repr__(self):
             '[%s]' % ', '.join('%s' % (item,) for item in self.items())
         )
 
+
+    __deepcopy__ = lambda self, memo: type(self)(
+                                               copy.deepcopy(self._dict, memo))
+        
+
 class _MutableBagMixin(_BaseBagMixin):
     # blocktodo: ensure all mutable methods, like __iadd__ and everything
     
diff --git a/source_py3/test_python_toolbox/test_nifty_collections/test_bagging.py b/source_py3/test_python_toolbox/test_nifty_collections/test_bagging.py
@@ -62,7 +62,8 @@ def test_common(self):
         
         assert self.bag_type(bag.elements()) == bag
         
-        assert +bag == bag
+        with cute_testing.RaiseAssertor(TypeError):
+            + bag
         with cute_testing.RaiseAssertor(TypeError):
             - bag
         
@@ -185,15 +186,16 @@ def test_ignores_zero(self):
             assert {bag_2, bag_3} == {bag_2} == {bag_3}
     
     def test_copy(self):
-        my_tuple = ('meow',)
-        bag = self.bag_type({my_tuple: 3, 'a': 4,})
+        class O: pass
+        o = O()
+        bag = self.bag_type({o: 3})
         bag_shallow_copy = copy.copy(bag)
         bag_deep_copy = copy.deepcopy(bag)
-        assert bag_shallow_copy == bag == bag_deep_copy
+        assert bag_shallow_copy == bag != bag_deep_copy
         assert next(iter(bag_shallow_copy)) == next(iter(bag_shallow_copy)) \
-                                                != next(iter(bag_shallow_copy))
+                                                   != next(iter(bag_deep_copy))
         assert next(iter(bag_shallow_copy)) is next(iter(bag_shallow_copy)) \
-                                            is not next(iter(bag_shallow_copy))
+                                               is not next(iter(bag_deep_copy))
         
         
         
