def test_get_angle_same(self): """ Test that when the same points are given to calculate the angle, an angle of 0 is returned. """ viz = drawable.Drawable(plt.figure(figsize=(10, 5))) graph = Graph(viz) self.assertEqual(0, round(graph._get_angle((1, 1), (1, 1)), 5))
def test_get_angle_different_dimensions(self): """ Test that when getting the angle between two points, the magnitude is normalized. """ viz = drawable.Drawable(plt.figure(figsize=(10, 5))) graph = Graph(viz) self.assertEqual(round(math.pi / 4., 5), round(graph._get_angle((1, 0), (2, 2)), 5)) self.assertEqual(round(math.pi / 2., 5), round(graph._get_angle((2, 0), (0, 1)), 5)) self.assertEqual(round(3 * math.pi / 4., 5), round(graph._get_angle((2, 0), (-1, 1)), 5)) self.assertEqual(round(math.pi, 5), round(graph._get_angle((1, 0), (-2, 0)), 5)) self.assertEqual(round(-3 * math.pi / 4, 5), round(graph._get_angle((2, 0), (-2, -2)), 5)) self.assertEqual(round(-math.pi / 2., 5), round(graph._get_angle((2, 0), (0, -1)), 5)) self.assertEqual(round(-math.pi / 4., 5), round(graph._get_angle((1, 0), (2, -2)), 5))
def test_get_angle(self): """ Test getting the angle between two points. """ viz = drawable.Drawable(plt.figure(figsize=(10, 5))) graph = Graph(viz) self.assertEqual(round(math.pi / 4., 5), round(graph._get_angle((1, 0), (1, 1)), 5)) self.assertEqual(round(math.pi / 2., 5), round(graph._get_angle((1, 0), (0, 1)), 5)) self.assertEqual(round(3 * math.pi / 4., 5), round(graph._get_angle((1, 0), (-1, 1)), 5)) self.assertEqual(round(math.pi, 5), round(graph._get_angle((1, 0), (-1, 0)), 5)) self.assertEqual(round(-3 * math.pi / 4, 5), round(graph._get_angle((1, 0), (-1, -1)), 5)) self.assertEqual(round(-math.pi / 2., 5), round(graph._get_angle((1, 0), (0, -1)), 5)) self.assertEqual(round(-math.pi / 4., 5), round(graph._get_angle((1, 0), (1, -1)), 5))