aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/PyHamcrest/hamcrest/library/number/ordering_comparison.py
blob: c3c75f425d6b7a69b9fa4c73e51e84565bb19ef7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from hamcrest.core.base_matcher import BaseMatcher
import operator

__author__ = "Jon Reid"
__copyright__ = "Copyright 2011 hamcrest.org"
__license__ = "BSD, see License.txt"


class OrderingComparison(BaseMatcher):

    def __init__(self, value, comparison_function, comparison_description):
        self.value = value
        self.comparison_function = comparison_function
        self.comparison_description = comparison_description

    def _matches(self, item):
        return self.comparison_function(item, self.value)

    def describe_to(self, description):
        description.append_text('a value ')                     \
                   .append_text(self.comparison_description)    \
                   .append_text(' ')                            \
                   .append_description_of(self.value)


def greater_than(value):
    """Matches if object is greater than a given value.

    :param value: The value to compare against.

    """
    return OrderingComparison(value, operator.gt, 'greater than')


def greater_than_or_equal_to(value):
    """Matches if object is greater than or equal to a given value.

    :param value: The value to compare against.

    """
    return OrderingComparison(value, operator.ge, 'greater than or equal to')


def less_than(value):
    """Matches if object is less than a given value.

    :param value: The value to compare against.

    """
    return OrderingComparison(value, operator.lt, 'less than')


def less_than_or_equal_to(value):
    """Matches if object is less than or equal to a given value.

    :param value: The value to compare against.

    """
    return OrderingComparison(value, operator.le, 'less than or equal to')