test_grid.py

import pytest
from grid import *

compare_list = lambda x, y: all([elm_x == y[i] for i, elm_x in enumerate(x)])


def test_distance():
    assert distance((1, 2), (5, 5)) == 5.0


def test_distance_squared():
    assert distance_squared((1, 2), (5, 5)) == 25.0


def test_clip():
    list_ = [clip(x, 0, 1) for x in [-1, 0.5, 10]]
    res = [0, 0.5, 1]

    assert compare_list(list_, res)


def test_vector_clip():
    assert vector_clip((-1, 10), (0, 0), (9, 9)) == (0, 9)

if __name__ == '__main__':
    pytest.main()