Enforce use of Decimal, formatting

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):
         """