1 files changed, 7 insertions, 12 deletions
diff --git a/finance/test_flow.py b/finance/test_flow.py
index db1b695..91d189b 100644
--- a/finance/test_flow.py
+++ b/finance/test_flow.py
@@ -31,22 +31,17 @@ def test_flow_integration() -> None:
 def test_simulate() -> None:
     flows = (Flow(amount=Decimal(100.0), since=None, until=None),)
 
-    dates, values = simulate(
+    simulated = simulate(
         start=datetime(2024, 1, 1),
         end=datetime(2024, 4, 1),
         flows=flows,
     )
 
-    assert dates == [
-        datetime(2024, 1, 1),
-        datetime(2024, 2, 1),
-        datetime(2024, 3, 1),
-        datetime(2024, 4, 1),
+    expected = [
+        (datetime(2024, 1, 1), Decimal(100.0)),
+        (datetime(2024, 2, 1), Decimal(200.0)),
+        (datetime(2024, 3, 1), Decimal(300.0)),
+        (datetime(2024, 4, 1), Decimal(400.0)),
     ]
 
-    assert values == [
-        Decimal(100.0),
-        Decimal(200.0),
-        Decimal(300.0),
-        Decimal(400.0),
-    ]
+    assert simulated == expected