Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/tests/test_utils.py
| # Copyright (C) 2018-2020 the Software Heritage developers | # Copyright (C) 2018-2020 the Software Heritage developers | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import pytest | import pytest | ||||
| from testing.postgresql import Postgresql | from testing.postgresql import Postgresql | ||||
| from swh.lister import utils | from swh.lister import utils | ||||
| @pytest.mark.parametrize( | @pytest.mark.parametrize( | ||||
| "total_pages,nb_pages,expected_ranges", | "total_pages,nb_pages,expected_ranges", | ||||
| [(14, 5, [(0, 5), (5, 10), (10, 14)]), (19, 10, [(0, 10), (10, 19)])], | [ | ||||
| (14, 5, [(0, 4), (5, 9), (10, 14)]), | |||||
| (19, 10, [(0, 9), (10, 19)]), | |||||
| (20, 3, [(0, 2), (3, 5), (6, 8), (9, 11), (12, 14), (15, 17), (18, 20)]), | |||||
| (21, 3, [(0, 2), (3, 5), (6, 8), (9, 11), (12, 14), (15, 17), (18, 21),],), | |||||
| ], | |||||
| ) | ) | ||||
| def test_split_range(total_pages, nb_pages, expected_ranges): | def test_split_range(total_pages, nb_pages, expected_ranges): | ||||
| actual_ranges = list(utils.split_range(total_pages, nb_pages)) | actual_ranges = list(utils.split_range(total_pages, nb_pages)) | ||||
| assert actual_ranges == expected_ranges | assert actual_ranges == expected_ranges | ||||
| @pytest.mark.parametrize("total_pages,nb_pages", [(None, 1), (100, None)]) | @pytest.mark.parametrize("total_pages,nb_pages", [(None, 1), (100, None)]) | ||||
| def test_split_range_errors(total_pages, nb_pages): | def test_split_range_errors(total_pages, nb_pages): | ||||
| Show All 15 Lines | |||||