import unittest

from fancymath.vector import Vector


class TestVector(unittest.TestCase):
    def test_add_vector(self):
        v1 = Vector(1, 2, 3)
        v2 = Vector(-3, -2, -1)
        v3 = v1 + v2
        self.assertIsInstance(v3, Vector)
        self.assertEqual(v3.x, v1.x + v2.x)

    def test_abs(self):
        v1 = Vector(0, 0, 0)
        v2 = Vector(1, 2, 2)
        v3 = Vector(1, -2, 2)
        self.assertEqual(v1.abs(), 0)
        self.assertEqual(v2.abs(), 3)
        self.assertEqual(v2.abs(), v3.abs())