[py3k] Fixed pagination_regress tests.

This commit is contained in:
Alex Gaynor 2012-08-16 10:54:45 -04:00
parent d69bd23b55
commit 37a894b48c
1 changed files with 4 additions and 2 deletions

View File

@ -17,14 +17,16 @@ class PaginatorTests(TestCase):
paginator = Paginator(*params)
self.check_attribute('count', paginator, count, params)
self.check_attribute('num_pages', paginator, num_pages, params)
self.check_attribute('page_range', paginator, page_range, params)
self.check_attribute('page_range', paginator, page_range, params, coerce=list)
def check_attribute(self, name, paginator, expected, params):
def check_attribute(self, name, paginator, expected, params, coerce=None):
"""
Helper method that checks a single attribute and gives a nice error
message upon test failure.
"""
got = getattr(paginator, name)
if coerce is not None:
got = coerce(got)
self.assertEqual(expected, got,
"For '%s', expected %s but got %s. Paginator parameters were: %s"
% (name, expected, got, params))