Python unittesting: проверьте, равны ли два угла равны

Question

Я хочу проверить функцию, которая выводит заголовок в градусах, который является числом в интервале [0, 360]. Поскольку результатом является число с плавающей запятой, сравнение фактического результата с ожидаемым с помощью unittest.assertEqual() не работает. unittest.assertAlmostEqual() лучше, поскольку он обеспечивает допуск. Этот подход работает для заголовка, который не близок к 0 градусам.

Вопрос: Каков правильный способ проверки заголовков, ожидаемое значение которых составляет 0 градусов? assertAlmostEquals() будут включать только углы, которые немного больше 0 градусов, но будут пропускать те, которые немного меньше 0, то есть 360 градусов ...

Answer 1

Вы можете использовать квадрат евклидова расстояния между двумя точками на единичном круге и закон косинусов для получения абсолютной разницы между двумя углами:

from math import sin, cos, acos 
from unittest import assertAlmostEqual   

def assertAlmostEqualAngles(x, y, **kwargs): 
    c2 = (sin(x)-sin(y))**2 + (cos(x)-cos(y))**2 
    angle_diff = acos((2.0 - c2)/2.0) # a = b = 1 
    assertAlmostEqual(angle_diff, 0.0, **kwargs) 

Это работает с радианами. Если угол в градусах, необходимо выполнить преобразование:

from math import sin, cos, acos, radians, degrees 
from unittest import assertAlmostEqual   

def assertAlmostEqualAngles(x, y, **kwargs): 
    x,y = radians(x),radians(y) 
    c2 = (sin(x)-sin(y))**2 + (cos(x)-cos(y))**2 
    angle_diff = degrees(acos((2.0 - c2)/2.0)) 
    assertAlmostEqual(angle_diff, 0.0, **kwargs)