day8/test_init.py


1
2
3
4
5
6
7

# -*- coding: utf-8 -*-
from day8 import AssignmentTwo


def test_calculate_1d_scenic_score():
    assert AssignmentTwo.calculate_1d_scenic_score([3, 5, 3, 5, 3], 1) == 1 * 2
    assert AssignmentTwo.calculate_1d_scenic_score([2, 5, 5, 1, 2], 2) == 1 * 2