implement Vector operators __add__, __sub__, __mul__, __truediv__

This commit is contained in:
Justin Buchanan
2018-11-18 21:33:17 -08:00
parent 7ac64a7280
commit 3c3837fef3
2 changed files with 25 additions and 0 deletions

View File

@ -108,6 +108,19 @@ class TestCadObjects(BaseTest):
result = Vector(1, 2, 0) + Vector(0, 0, 3)
self.assertTupleAlmostEquals((1.0, 2.0, 3.0), result.toTuple(), 3)
def testVectorOperators(self):
result = Vector(1, 1, 1) + Vector(2, 2, 2)
self.assertEqual(Vector(3, 3, 3), result)
result = Vector(1, 2, 3) - Vector(3, 2, 1)
self.assertEqual(Vector(-2, 0, 2), result)
result = Vector(1, 2, 3) * 2
self.assertEqual(Vector(2, 4, 6), result)
result = Vector(2, 4, 6) / 2
self.assertEqual(Vector(1, 2, 3), result)
def testVectorEquals(self):
a = Vector(1, 2, 3)
b = Vector(1, 2, 3)