-

cool-RR · cool-RR · commit b27bb2062c03 · 2014-09-26T21:14:09.000+03:00
diff --git a/source_py3/python_toolbox/math_tools/sequences.py b/source_py3/python_toolbox/math_tools/sequences.py
@@ -63,9 +63,9 @@ def abs_stirling(n, k):
 def shitfuck(k, recurrence_counter):
     from python_toolbox import nifty_collections
     from python_toolbox import cute_iter_tools
-    if not isinstance(recurrence_counter, nifty_collections.FrozenCounter):
+    if not isinstance(recurrence_counter, nifty_collections.FrozenChunkCounter):
         recurrence_counter = \
-                            nifty_collections.FrozenCounter(recurrence_counter)
+                       nifty_collections.FrozenChunkCounter(recurrence_counter)
     if k == 1:
         assert recurrence_counter # Works because `FrozenCounter` implements
                                   # `__bool__`.
@@ -79,7 +79,7 @@ def shitfuck(k, recurrence_counter):
     current_reccurence_counters = {recurrence_counter}
     while len(levels) < k and current_reccurence_counters:
         levels.append(
-            {recurrence_counter_: get_sub_counters_counter(recurrence_counter_)
+            {recurrence_counter_: recurrence_counter_.get_sub_counters_counter()
              for recurrence_counter_ in current_reccurence_counters
                                   if recurrence_counter_ not in _shitfuck_cache
         })
diff --git a/source_py3/python_toolbox/nifty_collections/frozen_chunk_counter.py b/source_py3/python_toolbox/nifty_collections/frozen_chunk_counter.py
@@ -10,7 +10,7 @@
 
 class FrozenChunkCounter(FrozenCounter):
     def __init__(self, iterable):
-        super().__init__()
+        super().__init__(iterable)
         
         # All zero values were already fileterd out by `FrozenCounter`, we'll
         # filter out just the non-natural-number keys.
@@ -23,8 +23,8 @@ def __init__(self, iterable):
             
     def get_sub_counters_counter(self):
         sub_counters_counter = collections.Counter()
-        for key_to_reduce, value_of_key_to_reduce in sub_counter.items():
-            sub_counter = collections.Counter(sub_counter)
+        for key_to_reduce, value_of_key_to_reduce in self.items():
+            sub_counter = collections.Counter(self)
             sub_counter[key_to_reduce] -= 1
             sub_counter[key_to_reduce - 1] += 1
             sub_counters_counter[FrozenChunkCounter(sub_counter)] = \
