kalpa/tests/common/test_angles.py

14 lines
380 B
Python

import numpy
import kalpaa.common.angles
def test_angles_in_range(snapshot):
angles_1 = numpy.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
angles_2 = numpy.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) * 4
result = kalpaa.common.angles.shortest_angular_distance(angles_1, angles_2)
assert (result >= -numpy.pi).all()
assert (result < numpy.pi).all()
assert result.tolist() == snapshot