pderaaij/python-fun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add addition and subtraction to Money and Price

Browse Source
This commit is contained in:
Paul de Raaij
2026-07-26 08:08:43 -04:00
parent 354ecbdd64
commit 3df59767ea
2 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
domain/marketplace/money.py
View File

@@ -5,3 +5,9 @@ from decimal import Decimal
@dataclass(frozen=True) @dataclass(frozen=True)
class Money: class Money:
amount: Decimal amount: Decimal
def __add__(self, other):
return Money(self.amount + other.amount)
def __sub__(self, other):
return Money(self.amount - other.amount)

15
tests/test_money.py
View File

@@ -22,3 +22,18 @@ def test_money_equality():
money1 = Money(amount=Decimal("50.00")) money1 = Money(amount=Decimal("50.00"))
money2 = Money(amount=Decimal("50.00")) money2 = Money(amount=Decimal("50.00"))
assert money1 == money2 assert money1 == money2
def test_money_add():
money1 = Money(amount=Decimal("2.00"))
money2 = Money(amount=Decimal("3.00"))
assert (money1 + money2) == Money(amount=Decimal("5.00"))
assert money1 == Money(amount=Decimal("2.00"))
assert money2 == Money(amount=Decimal("3.00"))
def test_money_substract():
money1 = Money(amount=Decimal("5.00"))
money2 = Money(amount=Decimal("3.00"))
assert (money1 - money2) == Money(amount=Decimal("2.00"))