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)