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