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))
