diff options
| -rw-r--r-- | test_trade_queue.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test_trade_queue.py b/test_trade_queue.py index 032f864..46e8c0d 100644 --- a/test_trade_queue.py +++ b/test_trade_queue.py @@ -136,20 +136,20 @@ class TestFIFOQueueRemove(unittest.TestCase): Set up a FIFOQueue instance with sample trades for testing. """ self.fifo_queue = FIFOQueue() - self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-03-17") - self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-17") - self.fifo_queue.add(Decimal("20"), Decimal("200"), "2025-04-18") - self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-18") + self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-03-17 12:00:00") + self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-17 12:15:00") + self.fifo_queue.add(Decimal("20"), Decimal("200"), "2025-04-18 13:20:00") + self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-18 14:50:00") # Duplicate date for testing def test_add_in_order(self): """ Test whether trades are ordered by date. """ - self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-01") + self.fifo_queue.add(Decimal("10"), Decimal("100"), "2025-04-01 11:11:00") lt = self.fifo_queue.get_copy() self.assertTrue( - all(lt[i].date <= lt[i+1].date for i in range(len(lt) - 1)) + all(lt[i].timestamp <= lt[i+1].timestamp for i in range(len(lt) - 1)) ) def test_remove_successful(self): |