blob: 691ca1a09eaad7726cfe1652e0affbadf82af66f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import unittest
from decimal import Decimal
from trade import 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")
def test_initialization(self):
"""
Test that the Trade instance initializes correctly.
"""
self.assertEqual(self.trade.amount, 10.0)
self.assertEqual(self.trade.total_cost, 100.0)
self.assertEqual(self.trade.date, "2025-04-14")
self.assertAlmostEqual(self.trade.price_per_coin, 10.0)
def test_remove_coins_valid(self):
"""
Test removing a valid amount of coins.
"""
self.trade.remove_coins(5.0)
self.assertEqual(self.trade.amount, 5.0)
self.assertAlmostEqual(self.trade.total_cost, 50.0)
self.assertAlmostEqual(self.trade.price_per_coin, 10.0)
def test_remove_coins_exceeds_amount(self):
"""
Test removing more coins than available.
"""
with self.assertRaises(ValueError):
self.trade.remove_coins(15.0)
def test_price_per_coin_division_by_zero(self):
"""
Test the price_per_coin property when the amount is zero.
"""
self.trade.remove_coins(10.0) # Reduce amount to zero
with self.assertRaises(ZeroDivisionError):
_ = self.trade.price_per_coin
def test_repr(self):
"""
Test the __repr__ method for correct string representation.
"""
expected_repr = "Trade(amount=10.0, price_per_coin=10.00, total_cost=100.00, date=2025-04-14)"
self.assertEqual(repr(self.trade), expected_repr)
if __name__ == "__main__":
unittest.main()
|
