1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# -*- 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
|
