test_all.py

import sys
import os
import numpy as np
import pytest

from frechetdist import frdist

TEST_CASES = [

    {
        'P': [[1, 1], [2, 1], [2, 2]],
        'Q': [[2, 2], [0, 1], [2, 4]],
        'expected': 2.0
    },

    {
        'P': np.array((np.linspace(0.0, 1.0, 100), np.ones(100))).T,
        'Q': np.array((np.linspace(0.0, 1.0, 100), np.ones(100))).T,
        'expected': 0
    },

    {
        'P': [[-1, 0], [0, 1], [1, 0], [0, -1]],
        'Q': [[-2, 0], [0, 2], [2, 0], [0, -2]],
        'expected': 1.0
    },

    {
        'P': np.array((np.linspace(0.0, 1.0, 100), np.ones(100)*2)).T,
        'Q': np.array((np.linspace(0.0, 1.0, 100), np.ones(100))).T,
        'expected': 1.0
    }

]


def test_main():
    for test_case in TEST_CASES:
        P = test_case['P']
        Q = test_case['Q']
        eo = test_case['expected']

        assert frdist(P, Q) == eo


def test_errors():

    P = []
    Q = [[2, 2], [0, 1], [2, 4]]

    with pytest.raises(ValueError):
        assert frdist(P, Q) == 2.0