summaryrefslogtreecommitdiffstats
path: root/day5/test_init.py
blob: a4d7bffbb027a7a368f0d298d7b3bf276e828a9b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from day5 import points_in_vector


def test_points_in_vector():
    assert points_in_vector(((1,1), (1,3))) == [
        (1, 1), (1, 2), (1, 3)
    ]

    assert points_in_vector(((9,7), (7,7))) == [
        (9, 7), (8, 7), (7, 7)
    ]

    assert points_in_vector(((1,1), (3,3)), False) == []

    assert points_in_vector(((1,1), (3,3)), True) == [
        (1, 1), (2, 2), (3, 3)
    ]

    assert points_in_vector(((9,7), (7,9)), False) == []

    assert points_in_vector(((9,7), (7,9)), True) == [
        (9, 7), (8, 8), (7, 9),
    ]