diff options
| -rw-r--r-- | test_trade_queue.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/test_trade_queue.py b/test_trade_queue.py index 692a433..e24e4e8 100644 --- a/test_trade_queue.py +++ b/test_trade_queue.py @@ -103,8 +103,12 @@ class TestFIFOQueue(unittest.TestCase): """ trades = self.queue.remove(5.0) # Remove 5 COIN from the first trade self.assertEqual(len(trades), 1) # Only one trade should be returned + + self.assertAlmostEqual(trades[0].price_per_coin, 10, 1) # Coin-cost needs to stay constant self.assertEqual(trades[0].amount, 5.0) # Check the removed amount self.assertEqual(trades[0].total_cost, 50.0) # Total cost should be proportional: (100 * 5 / 10) + + self.assertAlmostEqual(self.queue.queue[0].price_per_coin, 10, 1) # Original total cost remains unchanged self.assertEqual(self.queue.queue[0].amount, 5.0) # Remaining amount in the first trade should be updated self.assertEqual(self.queue.queue[0].total_cost, 50.0) # Original total cost remains unchanged |