Check price per coin

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