wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-30 06:28:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

TST: Adds assert_equal dispatch for slices

Browse Source
This commit is contained in:
Joe Jevnik
2016-10-07 18:57:05 -04:00
parent eba02da271
commit 8e26368d4e
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/testing/predicates.py
+26
View File
@@ -519,6 +519,32 @@ def assert_timestamp_and_datetime_equal(result,
)
@assert_equal.register(slice, slice)
def assert_slice_equal(result, expected, path=(), msg=''):
diff_start = (
('starts are not equal: %s != %s' % (result.start, result.stop))
if result.start != expected.start else
''
)
diff_stop = (
('stops are not equal: %s != %s' % (result.stop, result.stop))
if result.stop != expected.stop else
''
)
diff_step = (
('steps are not equal: %s != %s' % (result.step, result.stop))
if result.step != expected.step else
''
)
diffs = diff_start, diff_stop, diff_step
assert not any(diffs), '%s%s\n%s' % (
_fmt_msg(msg),
'\n'.join(filter(None, diffs)),
_fmt_path(path),
)
def assert_isidentical(result, expected, msg=''):
assert result.isidentical(expected), (
'%s%s is not identical to %s' % (_fmt_msg(msg), result, expected)