# -*- 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