diff --git a/tests/test_trivial.py b/tests/test_trivial.py index d9035751..7661dbea 100644 --- a/tests/test_trivial.py +++ b/tests/test_trivial.py @@ -1,7 +1,15 @@ import numpy as np +import pytest from regional_mom6 import angle_between # placeholder trivial test test -def test_angle_between(): - assert np.isclose(angle_between([1, 0, 0], [0, 1, 0], [0, 0, 1]), np.pi / 2) +@pytest.mark.parametrize( + ("v1", "v2", "v3", "true_angle"), + [ + ([1, 0, 0], [0, 1, 0], [0, 0, 1], np.pi / 2), + ([1, 0, 0], [1, 1, 0], [0, 1, 1], np.pi / 4), + ], +) +def test_angle_between(v1, v2, v3, true_angle): + assert np.isclose(angle_between(v1, v2, v3), true_angle)