-

cool-RR · cool-RR · commit 468d1f7cb599 · 2015-06-09T19:24:49.000+03:00
diff --git a/source_py2/test_python_toolbox/test_nifty_collections/test_bagging.py b/source_py2/test_python_toolbox/test_nifty_collections/test_bagging.py
@@ -1,6 +1,8 @@
 # Copyright 2009-2015 Ram Rachum.
 # This program is distributed under the MIT license.
 
+from __future__ import division
+
 import re
 import pickle
 import abc
@@ -454,11 +456,9 @@ def test_mutating(self):
         assert bag == self.bag_type('abracadabra' + 'a' * 5)
         assert bag is bag_reference
 
-        # We only allow floor division on bags, not regular divison, because a
-        # decimal bag is unheard of.
         bag = bag_reference = self.bag_type('abracadabra')
         with cute_testing.RaiseAssertor(TypeError):
-            bag['a'] /= 2
+            bag['a'] /= 2 # Won't work because `bag['a']` happens to be odd.
 
         bag = bag_reference = self.bag_type('abracadabra')
         bag['a'] //= 2
@@ -500,6 +500,12 @@ def test_mutating(self):
         assert bag == self.bag_type('abracadabra' * 2)
         assert bag is bag_reference
 
+        # We only allow floor division on bags, not regular divison, because a
+        # decimal bag is unheard of.
+        bag = bag_reference = self.bag_type('abracadabra')
+        with cute_testing.RaiseAssertor(TypeError):
+            bag /= 2
+
         bag = bag_reference = self.bag_type('abracadabra')
         bag //= 2
         assert bag == self.bag_type('aabr')
@@ -665,7 +671,7 @@ def test_mutating(self):
         # decimal bag is unheard of.
         bag = bag_reference
         with cute_testing.RaiseAssertor(TypeError):
-            bag['a'] /= 2
+            bag /= 2
         
         bag = bag_reference
         bag //= 3
diff --git a/source_py3/python_toolbox/logic_tools.py b/source_py3/python_toolbox/logic_tools.py
@@ -60,7 +60,7 @@ def all_equivalent(iterable, relation=operator.eq, *, assume_reflexive=True,
     return all(itertools.starmap(relation, pairs))
 
 
-def get_equivalence_classes(iterable, key=None, container=set, *
+def get_equivalence_classes(iterable, key=None, container=set, *, 
                             use_ordered_dict=False, sort_ordered_dict=False):
     '''
     Divide items in `iterable` to equivalence classes, using the key function.
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
@@ -453,11 +453,9 @@ def test_mutating(self):
         assert bag == self.bag_type('abracadabra' + 'a' * 5)
         assert bag is bag_reference
 
-        # We only allow floor division on bags, not regular divison, because a
-        # decimal bag is unheard of.
         bag = bag_reference = self.bag_type('abracadabra')
         with cute_testing.RaiseAssertor(TypeError):
-            bag['a'] /= 2
+            bag['a'] /= 2 # Won't work because `bag['a']` happens to be odd.
 
         bag = bag_reference = self.bag_type('abracadabra')
         bag['a'] //= 2
@@ -499,6 +497,12 @@ def test_mutating(self):
         assert bag == self.bag_type('abracadabra' * 2)
         assert bag is bag_reference
 
+        # We only allow floor division on bags, not regular divison, because a
+        # decimal bag is unheard of.
+        bag = bag_reference = self.bag_type('abracadabra')
+        with cute_testing.RaiseAssertor(TypeError):
+            bag /= 2
+
         bag = bag_reference = self.bag_type('abracadabra')
         bag //= 2
         assert bag == self.bag_type('aabr')
@@ -626,7 +630,7 @@ def test_mutating(self):
         with cute_testing.RaiseAssertor(TypeError):
             bag['a'] *= 2
         with cute_testing.RaiseAssertor(TypeError):
-            bag['a'] /= 2
+            bag['a'] /= 2 
         with cute_testing.RaiseAssertor(TypeError):
             bag['a'] //= 2
         with cute_testing.RaiseAssertor(TypeError):
@@ -664,7 +668,7 @@ def test_mutating(self):
         # decimal bag is unheard of.
         bag = bag_reference
         with cute_testing.RaiseAssertor(TypeError):
-            bag['a'] /= 2
+            bag /= 2
         
         bag = bag_reference
         bag //= 3
