diff --git a/test/test_vector.py b/test/test_vector.py index 3606b6d..53938f9 100644 --- a/test/test_vector.py +++ b/test/test_vector.py @@ -32,3 +32,16 @@ class TestVector(unittest.TestCase): v3 = v1 + v2 self.assertIsInstance(v3, Vector) self.assertEqual(v3.x, v1.x + v2.x) + + def test_dot_failure(self): + with self.assertRaises(ValueError): + v1 = Vector(1, 2, 3) + v1.dot(3) + + def test_dot_product(self): + v1 = Vector(1, 2, 3) + v2 = Vector(-3, -2, -1) + v3 = v1.dot(v2) + self.assertEqual(v3,v1.x * v2.x + v1.y * v2.y + v1.z * v2.z) + vzero = Vector(0, 0, 0) + self.assertEqual(v1.dot(vzero),0)