-

cool-RR · cool-RR · commit 55fa7ed3ba07 · 2016-01-12T21:57:35.000+02:00
diff --git a/source_py3/python_toolbox/math_tools/misc.py b/source_py3/python_toolbox/math_tools/misc.py
@@ -151,4 +151,33 @@ def is_integer(x):
     except (TypeError, ValueError, OverflowError):
         return False
     return inted_x == x
+    
+    
+def cute_round(x, step=1, up=False):
+    '''
+    Round with a chosen step.
+    
+    Examples:
+    
+        >>> cute_round(7.456)
+        7
+        >>> cute_round(7.456, up=True)
+        8
+        >>> cute_round(7.456, step=0.1)
+        7.4
+        >>> cute_round(7.456, step=0.1, up=True)
+        7.5
+        >>> cute_round(7.456, step=0.2)
+        7.4
+        >>> cute_round(7.456, step=0.2, up=True)
+        7.6
+        >>> cute_round(7.456, step=0.01)
+        7.45
+        >>> cute_round(7.456, step=0.01, up=True)
+        7.46
+        
+    '''
+    div, mod = divmod(x, step)
+    return (div + bool(mod and up)) * step
+    
     
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
@@ -0,0 +1,18 @@
+# Copyright 2009-2015 Ram Rachum.
+# This program is distributed under the MIT license.
+
+from python_toolbox.math_tools import cute_round
+
+def almost_equals(x, y):
+    return (abs(1-(x / y)) < (10 ** -10))
+    
+
+def test_cute_round():
+    assert almost_equals(cute_round(7.456), 7)
+    assert almost_equals(cute_round(7.456, up=True), 8)
+    assert almost_equals(cute_round(7.456, step=0.1), 7.4)
+    assert almost_equals(cute_round(7.456, step=0.1, up=True), 7.5)
+    assert almost_equals(cute_round(7.456, step=0.2), 7.4)
+    assert almost_equals(cute_round(7.456, step=0.2, up=True), 7.6)
+    assert almost_equals(cute_round(7.456, step=0.01), 7.45)
+    assert almost_equals(cute_round(7.456, step=0.01, up=True), 7.46)
