def test_angle_quadrants(self):
# Test each quadrant
v1 = Vector(1, 1)
v2 = Vector(-1, 1)
v3 = Vector(-1, -1)
v4 = Vector(1, -1)
self.assertEqual(v1.get_angle(), 45)
self.assertEqual(v2.get_angle(), 135)
self.assertEqual(v3.get_angle(), 225)
self.assertEqual(v4.get_angle(), 315)
def test_angle_straight(self):
# Test each straight
v1 = Vector(1, 0)
v2 = Vector(0, 1)
v3 = Vector(-1, 0)
v4 = Vector(0, -1)
self.assertEqual(v1.get_angle(), 0)
self.assertEqual(v2.get_angle(), 90)
self.assertEqual(v3.get_angle(), 180)
self.assertEqual(v4.get_angle(), 270)
def test_zero(self):
v = Vector(0.0, 0.0)
self.assertEqual(v.x, 0.0)
self.assertEqual(v.y, 0.0)
self.assertEqual(v.get_magnitude(), 0.0)
self.assertEqual(v.get_angle(), None)
def test_init(self):
v = Vector(1, 1)
self.assertEqual(v.x, 1.0)
self.assertEqual(v.y, 1.0)
# Type check
self.assertIsInstance(v.x, float)
self.assertIsInstance(v.y, float)
self.assertIsInstance(v.get_magnitude(), float)
self.assertIsInstance(v.get_angle(), float)
def test_angle(self):
v = Vector(3.0, 4.0)
self.assertEqual(v.get_angle(), np.degrees(np.arctan(4.0 / 3.0)))
