123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
from math import copysign, isnan
|
|
|
|
|
|
class ExtraAssertions:
|
|
|
|
def assertIsSubclass(self, cls, superclass, msg=None):
|
|
if issubclass(cls, superclass):
|
|
return
|
|
standardMsg = f'{cls!r} is not a subclass of {superclass!r}'
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertNotIsSubclass(self, cls, superclass, msg=None):
|
|
if not issubclass(cls, superclass):
|
|
return
|
|
standardMsg = f'{cls!r} is a subclass of {superclass!r}'
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertHasAttr(self, obj, name, msg=None):
|
|
if not hasattr(obj, name):
|
|
if isinstance(obj, types.ModuleType):
|
|
standardMsg = f'module {obj.__name__!r} has no attribute {name!r}'
|
|
elif isinstance(obj, type):
|
|
standardMsg = f'type object {obj.__name__!r} has no attribute {name!r}'
|
|
else:
|
|
standardMsg = f'{type(obj).__name__!r} object has no attribute {name!r}'
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertNotHasAttr(self, obj, name, msg=None):
|
|
if hasattr(obj, name):
|
|
if isinstance(obj, types.ModuleType):
|
|
standardMsg = f'module {obj.__name__!r} has unexpected attribute {name!r}'
|
|
elif isinstance(obj, type):
|
|
standardMsg = f'type object {obj.__name__!r} has unexpected attribute {name!r}'
|
|
else:
|
|
standardMsg = f'{type(obj).__name__!r} object has unexpected attribute {name!r}'
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertStartsWith(self, s, prefix, msg=None):
|
|
if s.startswith(prefix):
|
|
return
|
|
standardMsg = f"{s!r} doesn't start with {prefix!r}"
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertNotStartsWith(self, s, prefix, msg=None):
|
|
if not s.startswith(prefix):
|
|
return
|
|
self.fail(self._formatMessage(msg, f"{s!r} starts with {prefix!r}"))
|
|
|
|
def assertEndsWith(self, s, suffix, msg=None):
|
|
if s.endswith(suffix):
|
|
return
|
|
standardMsg = f"{s!r} doesn't end with {suffix!r}"
|
|
self.fail(self._formatMessage(msg, standardMsg))
|
|
|
|
def assertNotEndsWith(self, s, suffix, msg=None):
|
|
if not s.endswith(suffix):
|
|
return
|
|
self.fail(self._formatMessage(msg, f"{s!r} ends with {suffix!r}"))
|
|
|
|
|
|
class ExceptionIsLikeMixin:
|
|
def assertExceptionIsLike(self, exc, template):
|
|
"""
|
|
Passes when the provided `exc` matches the structure of `template`.
|
|
Individual exceptions don't have to be the same objects or even pass
|
|
an equality test: they only need to be the same type and contain equal
|
|
`exc_obj.args`.
|
|
"""
|
|
if exc is None and template is None:
|
|
return
|
|
|
|
if template is None:
|
|
self.fail(f"unexpected exception: {exc}")
|
|
|
|
if exc is None:
|
|
self.fail(f"expected an exception like {template!r}, got None")
|
|
|
|
if not isinstance(exc, ExceptionGroup):
|
|
self.assertEqual(exc.__class__, template.__class__)
|
|
self.assertEqual(exc.args[0], template.args[0])
|
|
else:
|
|
self.assertEqual(exc.message, template.message)
|
|
self.assertEqual(len(exc.exceptions), len(template.exceptions))
|
|
for e, t in zip(exc.exceptions, template.exceptions):
|
|
self.assertExceptionIsLike(e, t)
|
|
|
|
|
|
class FloatsAreIdenticalMixin:
|
|
def assertFloatsAreIdentical(self, x, y):
|
|
"""Fail unless floats x and y are identical, in the sense that:
|
|
(1) both x and y are nans, or
|
|
(2) both x and y are infinities, with the same sign, or
|
|
(3) both x and y are zeros, with the same sign, or
|
|
(4) x and y are both finite and nonzero, and x == y
|
|
|
|
"""
|
|
msg = 'floats {!r} and {!r} are not identical'
|
|
|
|
if isnan(x) or isnan(y):
|
|
if isnan(x) and isnan(y):
|
|
return
|
|
elif x == y:
|
|
if x != 0.0:
|
|
return
|
|
# both zero; check that signs match
|
|
elif copysign(1.0, x) == copysign(1.0, y):
|
|
return
|
|
else:
|
|
msg += ': zeros have different signs'
|
|
self.fail(msg.format(x, y))
|
|
|
|
|
|
class ComplexesAreIdenticalMixin(FloatsAreIdenticalMixin):
|
|
def assertComplexesAreIdentical(self, x, y):
|
|
"""Fail unless complex numbers x and y have equal values and signs.
|
|
|
|
In particular, if x and y both have real (or imaginary) part
|
|
zero, but the zeros have different signs, this test will fail.
|
|
|
|
"""
|
|
self.assertFloatsAreIdentical(x.real, y.real)
|
|
self.assertFloatsAreIdentical(x.imag, y.imag)
|