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),
    ]