31 lines
906 B
Python
31 lines
906 B
Python
"""
|
|
Assertion helpers for offsets tests
|
|
"""
|
|
|
|
|
|
def assert_offset_equal(offset, base, expected):
|
|
actual = offset + base
|
|
actual_swapped = base + offset
|
|
actual_apply = offset.apply(base)
|
|
try:
|
|
assert actual == expected
|
|
assert actual_swapped == expected
|
|
assert actual_apply == expected
|
|
except AssertionError:
|
|
raise AssertionError(
|
|
"\nExpected: {expected}\nActual: {actual}\nFor Offset: {offset})"
|
|
"\nAt Date: {base}".format(
|
|
expected=expected, actual=actual, offset=offset, base=base
|
|
)
|
|
)
|
|
|
|
|
|
def assert_onOffset(offset, date, expected):
|
|
actual = offset.onOffset(date)
|
|
assert actual == expected, (
|
|
"\nExpected: {expected}\nActual: {actual}\nFor Offset: {offset})"
|
|
"\nAt Date: {date}".format(
|
|
expected=expected, actual=actual, offset=offset, date=date
|
|
)
|
|
)
|