From d9321f4795f30819f8b7ef8f41e8fddd3ac17885 Mon Sep 17 00:00:00 2001 From: Joe Jevnik Date: Wed, 1 Feb 2017 19:10:20 -0500 Subject: [PATCH] TST: assert_equal dispatch for tuples --- zipline/testing/predicates.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/zipline/testing/predicates.py b/zipline/testing/predicates.py index cf4f042e..fa776ba1 100644 --- a/zipline/testing/predicates.py +++ b/zipline/testing/predicates.py @@ -29,7 +29,6 @@ from nose.tools import ( # noqa assert_raises, assert_raises_regexp, assert_regexp_matches, - assert_sequence_equal, assert_true, assert_tuple_equal, ) @@ -371,12 +370,14 @@ def assert_dict_equal(result, expected, path=(), msg='', **kwargs): @assert_equal.register(list, list) -def assert_list_equal(result, expected, path=(), msg='', **kwargs): +@assert_equal.register(tuple, tuple) +def assert_sequence_equal(result, expected, path=(), msg='', **kwargs): result_len = len(result) expected_len = len(expected) assert result_len == expected_len, ( - '%slist lengths do not match: %d != %d\n%s' % ( + '%s%s lengths do not match: %d != %d\n%s' % ( _fmt_msg(msg), + type(result).__name__, result_len, expected_len, _fmt_path(path),