def test_calculat_angles_for_multiple_centers(self): p1 = [0, 1] p2 = [1, 0] c = np.linspace(-0.5, 0.5) angles = calculate_angles(p1, p2, c) self.assertAlmostEqual(np.sum(angles), np.pi/2*len(c))
def test_calculate_angles_for_single_values(self): p1 = [0, 1] p2 = [1, 0] c = [0, 0] angle = calculate_angles(p1, p2, c) self.assertAlmostEqual(angle, np.pi/2)