ydb/contrib/python/PyHamcrest/py2/hamcrest/library/number/ordering_comparison.py

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')