Skip to content

Commit a61fb87

Browse files
ndawelarsmans
authored andcommitted
utils.testing: add assert_greater_equal and assert_less_equal
1 parent 7f72713 commit a61fb87

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

sklearn/utils/testing.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,9 @@
5454
__all__ = ["assert_equal", "assert_not_equal", "assert_raises",
5555
"assert_raises_regexp", "raises", "with_setup", "assert_true",
5656
"assert_false", "assert_almost_equal", "assert_array_equal",
57-
"assert_array_almost_equal", "assert_array_less"]
57+
"assert_array_almost_equal", "assert_array_less",
58+
"assert_less", "assert_less_equal",
59+
"assert_greater", "assert_greater_equal"]
5860

5961

6062
try:
@@ -105,6 +107,20 @@ def _assert_greater(a, b, msg=None):
105107
assert a > b, message
106108

107109

110+
def assert_less_equal(a, b, msg=None):
111+
message = "%r is not lower than or equal to %r" % (a, b)
112+
if msg is not None:
113+
message += ": " + msg
114+
assert a <= b, message
115+
116+
117+
def assert_greater_equal(a, b, msg=None):
118+
message = "%r is not greater than or equal to %r" % (a, b)
119+
if msg is not None:
120+
message += ": " + msg
121+
assert a >= b, message
122+
123+
108124
# To remove when we support numpy 1.7
109125
def assert_warns(warning_class, func, *args, **kw):
110126
"""Test that a certain warning occurs.

0 commit comments

Comments
 (0)