From f648b5a05a1dceeb000806ca64dd33c3980ee550 Mon Sep 17 00:00:00 2001 From: uvok Date: Mon, 14 Apr 2025 19:56:06 +0200 Subject: Check price per coin --- test_trade_queue.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'test_trade_queue.py') 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 -- cgit v1.2.3