diff options
Diffstat (limited to 'test_trade.py')
| -rw-r--r-- | test_trade.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test_trade.py b/test_trade.py index ab853d1..8cbf0f4 100644 --- a/test_trade.py +++ b/test_trade.py @@ -9,7 +9,7 @@ class TestTrade(unittest.TestCase): """ Set up a Trade instance for testing. """ - self.trade = Trade(amount=Decimal(10.0), total_cost=Decimal(100.0), date="2025-04-14") + self.trade = Trade(amount=Decimal(10.0), total_cost=Decimal(100.0), date="2025-04-14", refid="abcd1234") def test_initialization(self): """ @@ -67,7 +67,7 @@ class TestTrade(unittest.TestCase): """ Test the __repr__ method for correct string representation. """ - expected_repr = "Trade(amount=10.00, price_per_coin=10.00, total_cost=100.00, date=2025-04-14)" + expected_repr = "Trade(amount=10.00, price_per_coin=10.00, total_cost=100.00, date=2025-04-14, refid=abcd1234)" self.assertEqual(repr(self.trade), expected_repr) |