feat: ✨ add __rmul__ to relativedelta

olliemath · Aug 18, 2023 · d2a960f · d2a960f
1 parent 95bac5c
commit d2a960f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/test_relativedelta.py b/tests/test_relativedelta.py
@@ -27,6 +27,7 @@ def test_duration_arithmetic():
 
     assert x // 2 == relativedelta(months=5 * 6 + 3, seconds=50)
     assert x * 2 == relativedelta(months=10 * 12 + 14, seconds=200)
+    assert 2 * x == x * 2, "Multiplication should be symmetric"
 
     # Dividing a 1-month by e.g. 2 has no unambiguous meaning
     with pytest.raises(TypeError, match="unsupported operand"):

diff --git a/urelativedelta/relativedelta.py b/urelativedelta/relativedelta.py
@@ -98,6 +98,9 @@ def __rsub__(self, other):
     def __mul__(self, n: int) -> RelativeDelta:
         return self.__class__(months=self.months * n, timedelta=self.timedelta * n)
 
+    def __rmul__(self, n: int) -> RelativeDelta:
+        return self * n
+
     def __floordiv__(self, n: int) -> RelativeDelta:
         return self.__class__(months=self.months // n, timedelta=self.timedelta // n)