|
| 1 | +import pytest |
| 2 | +from vortex_utils.python_utils import * |
| 3 | + |
| 4 | +def test_ssa(): |
| 5 | + assert ssa(0) == 0 |
| 6 | + assert ssa(2 * np.pi) == 0 |
| 7 | + assert ssa(3.5) == pytest.approx(-2.78, rel=0.01) |
| 8 | + assert ssa(-3.5) == pytest.approx(2.78, rel=0.01) |
| 9 | + |
| 10 | +def test_euler_to_quat(): |
| 11 | + quat = euler_to_quat(0, 0, 0) |
| 12 | + assert quat == pytest.approx([0, 0, 0, 1], rel=0.01) |
| 13 | + quat = euler_to_quat(1, 0, 0) |
| 14 | + assert quat == pytest.approx([0.479, 0, 0, 0.877], rel=0.01) |
| 15 | + quat = euler_to_quat(0, 1, 0) |
| 16 | + assert quat == pytest.approx([0, 0.479, 0, 0.877], rel=0.01) |
| 17 | + quat = euler_to_quat(0, 0, 1) |
| 18 | + assert quat == pytest.approx([0, 0, 0.479, 0.877], rel=0.01) |
| 19 | + quat = euler_to_quat(1, 1, 1) |
| 20 | + assert quat == pytest.approx([0.5709, 0.167, 0.5709, 0.565], rel=0.01) |
| 21 | + |
0 commit comments