-

cool-RR · cool-RR · commit 8ac963c5bcfd · 2016-04-20T23:29:59.000+03:00
diff --git a/source_py3/python_toolbox/math_tools/misc.py b/source_py3/python_toolbox/math_tools/misc.py
@@ -168,7 +168,7 @@ def cute_round(x, round_mode=RoundMode.CLOSEST_OR_DOWN, *, step=1):
     Round with a chosen step.
     
     Examples:
-    blocktododoc
+    blocktododoc explain the different options, especially probabilistic
         >>> cute_round(7.456)
         7
         >>> cute_round(7.456, up=True)
diff --git a/source_py3/test_python_toolbox/test_math_tools/test_cute_round.py b/source_py3/test_python_toolbox/test_math_tools/test_cute_round.py
@@ -3,6 +3,8 @@
 
 import inspect
 
+from python_toolbox import nifty_collections
+
 from python_toolbox.math_tools import cute_round, RoundMode
 
 def almost_equals(x, y):
@@ -20,3 +22,86 @@ def test_closest_or_down(self):
         assert almost_equals(cute_round(7.456, step=0.1), 7.5)
         assert almost_equals(cute_round(7.456, step=0.2), 7.4)
         assert almost_equals(cute_round(7.456, step=0.01), 7.46)
+        
+    def test_closest_or_up(self):
+        assert almost_equals(
+            cute_round(7.456, RoundMode.CLOSEST_OR_UP), 7
+        )
+        assert almost_equals(
+            cute_round(7.654, RoundMode.CLOSEST_OR_UP), 8
+        )
+        assert almost_equals(
+            cute_round(7.5, RoundMode.CLOSEST_OR_UP), 8
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.CLOSEST_OR_UP, step=0.1), 7.5
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.CLOSEST_OR_UP, step=0.2), 7.4
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.CLOSEST_OR_UP, step=0.01), 7.46
+        )
+        
+    def test_always_up(self):
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_UP), 8
+        )
+        assert almost_equals(
+            cute_round(7.654, RoundMode.ALWAYS_UP), 8
+        )
+        assert almost_equals(
+            cute_round(7.5, RoundMode.ALWAYS_UP), 8
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_UP, step=0.1), 7.5
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_UP, step=0.2), 7.6
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_UP, step=0.01), 7.46
+        )
+        
+    def test_always_down(self):
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_DOWN), 7
+        )
+        assert almost_equals(
+            cute_round(7.654, RoundMode.ALWAYS_DOWN), 7
+        )
+        assert almost_equals(
+            cute_round(7.5, RoundMode.ALWAYS_DOWN), 7
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_DOWN, step=0.1), 7.4
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_DOWN, step=0.2), 7.4
+        )
+        assert almost_equals(
+            cute_round(7.456, RoundMode.ALWAYS_DOWN, step=0.01), 7.45
+        )
+        
+    def test_probabilistic(self):
+        def get_bag(*args, **kwargs):
+            kwargs.update({'round_mode': RoundMode.PROBABILISTIC,})
+            return nifty_collections.Bag(
+                cute_round(*args, **kwargs) for i in range(1000)
+            )
+            
+        bag = get_bag(5, step=5)
+        assert bag[5] == 1000
+        
+        bag = get_bag(6, step=5)
+        assert 300 <= bag[5] <= 908
+        assert 2 <= bag[10] <= 600
+        
+        bag = get_bag(7.5, step=5)
+        assert 100 <= bag[5] <= 900
+        assert 100 <= bag[10] <= 900
+        
+        bag = get_bag(10, step=5)
+        assert bag[10] == 1000
+                
+            

Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ def cute_round(x, round_mode=RoundMode.CLOSEST_OR_DOWN, *, step=1):`
`168`	`168`	`Round with a chosen step.`
`169`	`169`
`170`	`170`	`Examples:`
`171`		`- blocktododoc`
	`171`	`+ blocktododoc explain the different options, especially probabilistic`
`172`	`172`	`>>> cute_round(7.456)`
`173`	`173`	`7`
`174`	`174`	`>>> cute_round(7.456, up=True)`