mirror of
https://github.com/neogeek23/drawshare.git
synced 2026-02-04 11:08:21 +00:00
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import six
|
|
from hamcrest.core.base_matcher import BaseMatcher
|
|
from math import fabs
|
|
|
|
__author__ = "Jon Reid"
|
|
__copyright__ = "Copyright 2011 hamcrest.org"
|
|
__license__ = "BSD, see License.txt"
|
|
|
|
|
|
def isnumeric(value):
|
|
"""Confirm that 'value' can be treated numerically; duck-test accordingly
|
|
"""
|
|
if isinstance(value, (float, complex) + six.integer_types):
|
|
return True
|
|
|
|
try:
|
|
_ = (fabs(value) + 0 - 0) * 1
|
|
return True
|
|
except ArithmeticError:
|
|
return True
|
|
except:
|
|
return False
|
|
return False
|
|
|
|
|
|
class IsCloseTo(BaseMatcher):
|
|
|
|
def __init__(self, value, delta):
|
|
if not isnumeric(value):
|
|
raise TypeError('IsCloseTo value must be numeric')
|
|
if not isnumeric(delta):
|
|
raise TypeError('IsCloseTo delta must be numeric')
|
|
|
|
self.value = value
|
|
self.delta = delta
|
|
|
|
def _matches(self, item):
|
|
if not isnumeric(item):
|
|
return False
|
|
return fabs(item - self.value) <= self.delta
|
|
|
|
def describe_mismatch(self, item, mismatch_description):
|
|
if not isnumeric(item):
|
|
super(IsCloseTo, self).describe_mismatch(item, mismatch_description)
|
|
else:
|
|
actual_delta = fabs(item - self.value)
|
|
mismatch_description.append_description_of(item) \
|
|
.append_text(' differed by ') \
|
|
.append_description_of(actual_delta)
|
|
|
|
def describe_to(self, description):
|
|
description.append_text('a numeric value within ') \
|
|
.append_description_of(self.delta) \
|
|
.append_text(' of ') \
|
|
.append_description_of(self.value)
|
|
|
|
|
|
def close_to(value, delta):
|
|
"""Matches if object is a number close to a given value, within a given
|
|
delta.
|
|
|
|
:param value: The value to compare against as the expected value.
|
|
:param delta: The maximum delta between the values for which the numbers
|
|
are considered close.
|
|
|
|
This matcher compares the evaluated object against ``value`` to see if the
|
|
difference is within a positive ``delta``.
|
|
|
|
Example::
|
|
|
|
close_to(3.0, 0.25)
|
|
|
|
"""
|
|
return IsCloseTo(value, delta)
|