Added '*' operator support for cq.Vector where cq.Vector is the right operand

This commit is contained in:
AGD
2021-07-26 10:02:09 +08:00
parent 1688a82b6b
commit 2fbb68f4c3
2 changed files with 6 additions and 0 deletions

View File

@ -155,6 +155,9 @@ class Vector(object):
def __truediv__(self, denom: float) -> "Vector": def __truediv__(self, denom: float) -> "Vector":
return self.multiply(1.0 / denom) return self.multiply(1.0 / denom)
def __rmul__(self, scale: float) -> "Vector":
return self.multiply(scale)
def normalized(self) -> "Vector": def normalized(self) -> "Vector":
"""Return a normalized version of this vector""" """Return a normalized version of this vector"""
return Vector(self.wrapped.Normalized()) return Vector(self.wrapped.Normalized())

View File

@ -244,6 +244,9 @@ class TestCadObjects(BaseTest):
result = Vector(1, 2, 3) * 2 result = Vector(1, 2, 3) * 2
self.assertEqual(Vector(2, 4, 6), result) self.assertEqual(Vector(2, 4, 6), result)
result = 3 * Vector(1, 2, 3)
self.assertEqual(Vector(3, 6, 9), result)
result = Vector(2, 4, 6) / 2 result = Vector(2, 4, 6) / 2
self.assertEqual(Vector(1, 2, 3), result) self.assertEqual(Vector(1, 2, 3), result)