diff --git a/test/test_vector.py b/test/test_vector.py index 5238172..4af67c9 100644 --- a/test/test_vector.py +++ b/test/test_vector.py @@ -5,18 +5,22 @@ from fancymath.vector import Vector class TestVector(unittest.TestCase): def test_cross_product(self): - # given - v1 = Vector(1, 2, 3) - v2 = Vector(3, 2, 1) + # Dictionary of (v1, v2): v3 + values = { + (Vector(1, 2, 3), Vector(3, 2, 1)): Vector(-4, 8, -4), + (Vector(0, 0, 0), Vector(1, 2, 3)): Vector(0, 0, 0), + (Vector(1, 0, 0), Vector(1, 0, 0)): Vector(0, 0, 0), + } - # when - v3 = v1 * v2 + for inputs, expected in values.items(): + v1, v2 = inputs + with self.subTest(v1=v1, v2=v2): + v3 = v1 * v2 - # then - self.assertIsInstance(v3, Vector) - self.assertEqual(v3.x, -4) - self.assertEqual(v3.y, 8) - self.assertEqual(v3.z, -4) + self.assertIsInstance(v3, Vector) + self.assertEqual(v3.x, expected.x) + self.assertEqual(v3.y, expected.y) + self.assertEqual(v3.z, expected.z) def test_cross_product_failure(self): v1 = Vector(1, 2, 3) @@ -40,7 +44,7 @@ class TestVector(unittest.TestCase): self.assertEqual(v1.abs(), 0) self.assertEqual(v2.abs(), 3) self.assertEqual(v2.abs(), v3.abs()) - + def test_dot_failure(self): with self.assertRaises(ValueError): v1 = Vector(1, 2, 3)