diff options
Diffstat (limited to 'test_trade.py')
| -rw-r--r-- | test_trade.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test_trade.py b/test_trade.py index c3d7764..ab853d1 100644 --- a/test_trade.py +++ b/test_trade.py @@ -3,12 +3,13 @@ from decimal import Decimal from trade import PriceAdaption, Trade + class TestTrade(unittest.TestCase): def setUp(self) -> None: """ Set up a Trade instance for testing. """ - self.trade = Trade(amount=10.0, total_cost=100.0, date="2025-04-14") + self.trade = Trade(amount=Decimal(10.0), total_cost=Decimal(100.0), date="2025-04-14") def test_initialization(self): """ @@ -45,7 +46,7 @@ class TestTrade(unittest.TestCase): self.assertEqual(self.trade.amount, 6.0) self.assertEqual(self.trade.total_cost, 100.0) - self.assertEqual(self.trade.price_per_coin, Decimal("100.0")/Decimal("6.0")) + self.assertEqual(self.trade.price_per_coin, Decimal("100.0") / Decimal("6.0")) def test_remove_coins_exceeds_amount(self): """ |