# -*- coding: utf-8 -*-
import pytest

from day3 import AssignmentOne, AssignmentTwo, Assignment


class TestAssignment:
    data = [
        ["987654321111111", 2, 98],
        ["811111111111119", 2, 89],
        ["234234234234278", 2, 78],
        ["818181911112111", 2, 92],
        ["987654321111111", 12, 987654321111],
        ["811111111111119", 12, 811111111119],
        ["234234234234278", 12, 434234234278],
        ["818181911112111", 12, 888911112111],
    ]

    @pytest.mark.parametrize("battery,n,joltage", data)
    def test_find_highest_joltage(self, battery: str, n: int, joltage: int):
        banks = next(Assignment.parse_item(battery))
        assert Assignment.find_highest_joltage(banks, n) == joltage