-

cool-RR · cool-RR · commit 45d82eab5677 · 2014-10-26T21:10:18.000+02:00
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,30 +365,30 @@ 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
+    def test_get_contained_bags(self):
+        bag = self.bag_type('abracadabra')
+        contained_bags = bag.get_contained_bags()
+        assert len(contained_bags) == 6 * 3 * 2 * 2 * 3
+        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.keys()),
+                    key=tuple(bag.keys()).index
+                )
             
+        contained_bags_tuple = tuple(contained_bags)
+        assert self.bag_type('abraca') in contained_bags_tuple
+        assert self.bag_type('bd') 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):
