diff --git a/tests/codewars/test_pagination_helper.py b/tests/codewars/test_pagination_helper.py new file mode 100644 index 0000000..12f3185 --- /dev/null +++ b/tests/codewars/test_pagination_helper.py @@ -0,0 +1,27 @@ +from src.codewars.kata_pagination_helper import PaginationHelper + + +def test_pagination(): + collection = ["a", "b", "c", "d", "e", "f"] + helper = PaginationHelper(collection, 4) + + assert helper.page_count() == 2 + assert helper.item_count() == 6 + assert helper.page_item_count(0) == 4 + assert helper.page_item_count(1) == 2 + assert helper.page_item_count(2) == -1 + assert helper.page_index(5) == 1 + assert helper.page_index(2) == 0 + assert helper.page_index(20) == -1 + assert helper.page_index(-10) == -1 + + empty = PaginationHelper([], 10) + + assert empty.item_count() == 0 + assert empty.page_count() == 0 + assert empty.page_index(0) == -1 + assert empty.page_index(1) == -1 + assert empty.page_index(-1) == -1 + assert empty.page_item_count(0) == -1 + assert empty.page_item_count(1) == -1 + assert empty.page_item_count(-1) == -1