-

cool-RR · cool-RR · commit 7747fb6bbdaf · 2014-10-20T02:20:24.000+03:00
diff --git a/source_py3/python_toolbox/nifty_collections/bagging.py b/source_py3/python_toolbox/nifty_collections/bagging.py
@@ -517,6 +517,27 @@ def __repr__(self):
     def __reversed__(self):
         # Gets overridden in `_OrderedBagMixin`.
         raise TypeError("Can't reverse an unordered bag.")
+    
+
+    def get_contained_bags(self):
+        '''
+        Get all counters that are subsets of this bags.
+        
+        This means all counters that have counts identical or smaller for each
+        key.
+        '''
+        from python_toolbox import combi
+        
+        keys, amounts = zip(*((key, amount) for key, amount in self.items()))
+        
+        return combi.MapSpace(
+            lambda amounts_tuple:
+                         type(self)(self._dict_type(zip(keys, amounts_tuple))),
+            combi.ProductSpace(map(lambda amount: range(amount+1), amounts))
+        )
+    
+
+    
         
         
 
@@ -791,7 +812,22 @@ def frozen_bag_bag(self):
     def get_mutable(self):
         '''Get a mutable version of this bag.'''
         return self._mutable_type(self)
-    
+
+    # Poor man's caching done here because we can't import
+    # `python_toolbox.caching` due to import loop:
+    _contained_bags = None
+    def get_contained_bags(self):
+        '''
+        Get all counters that are subsets of this bags.
+        
+        This means all counters that have counts identical or smaller for each
+        key.
+        '''
+        if self._contained_bags is None:
+            self._contained_bags = super().get_contained_bags()
+        return self._contained_bags
+        
+
 
 class _BaseDictDelegator(collections.MutableMapping):
     '''
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
@@ -365,6 +365,31 @@ def test_operations(self):
         
         
         
+        def test_get_contained_bags(self):
+            bag = self.bag_type('abracadabra')
+            contained_bags = bag.get_contained_bags()
+            assert len(contained_bags) == 2 ** len('abracadabra')
+            had_full_one = False
+            for contained_bag in contained_bags:
+                assert contained_bag <= bag
+                if contained_bag == bag:
+                    assert had_full_one is False
+                    had_full_one = True
+                else:
+                    assert contained_bag < bag
+                if isinstance(bag, nifty_collections.Ordered):
+                    assert cute_iter_tools.is_sorted(
+                        tuple(contained_bag.items()),
+                        key=tuple(bag.items()).index
+                    )
+                
+            contained_bags_tuple = tuple(contained_bags)
+            assert self.bag_type('arcaba') in contained_bags_tuple
+            assert self.bag_type('db') in contained_bags_tuple
+            assert self.bag_type() in contained_bags_tuple
+            assert self.bag_type('x') not in contained_bags_tuple
+            
+        
         
 class BaseMutableBagTestCase(BaseBagTestCase):
     
