def test_paginator_with_more_than_10_pages_should_hide_some_of_them(self):
results = list(range(0, 300))
results_per_page = 20
paginator = Paginator(results, results_per_page)
paginator.page(1)
self.assertEqual(paginator.paginate_sections(),
[1, 2, 3, 4, 5, None, 11, 12, 13, 14, 15])
def test_paginator_with_less_than_10_pages_should_return_all_pages(self):
results = list(range(0, 199))
results_per_page = 20
paginator = Paginator(results, results_per_page)
paginator.page(1)
self.assertEqual(paginator.paginate_sections(),
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
