summaryrefslogtreecommitdiffstats
path: root/day5/test_init.py
blob: 018387b6633c3ccaca8cd3cc339e3be1756f954e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# -*- coding: utf-8 -*-
from unittest import result

import pytest

from day5 import AssignmentTwo


class TestAssignmentTwo:
    data = [
        [
            [range(3, 6), range(10, 15), range(16, 21), range(12, 19)],
            [range(3, 6), range(10, 21)],
        ],
        [[range(2, 8), range(4, 6)], [range(2, 8)]],
        [[range(4, 6), range(2, 8)], [range(2, 8)]],
        [[range(1, 3), range(3, 6)], [range(1, 6)]],
        [[range(3, 6), range(1, 3)], [range(1, 6)]],
    ]

    @pytest.mark.parametrize("ranges,result", data)
    def test_merge_ranges(self, ranges: list[range], result: list[range]):
        assert AssignmentTwo.merge_ranges(ranges) == result