Add measured values

2 files changed, 26 insertions, 6 deletions

diff --git a/demo b/demo
index 1ab3a18..db48ea7 100755
--- a/demo
+++ b/demo
@@ -16,6 +16,12 @@ DESCRIPTION = "Demo application plotting financial data with finance-py"
 def main() -> None:
     argparse.ArgumentParser(description=DESCRIPTION).parse_args()
 
+    measured = (
+        (datetime(2024, 1, 1), Decimal(105.0)),
+        (datetime(2024, 2, 1), Decimal(207.0)),
+        (datetime(2024, 3, 1), Decimal(334.0)),
+    )
+
     flows = (
         Flow(amount=Decimal(100.0), since=None, until=None),
         Flow(
@@ -25,13 +31,16 @@ def main() -> None:
         ),
     )
 
-    data = simulate(
+    simulated = simulate(
         start=datetime(2024, 1, 1),
         end=datetime(2030, 1, 1),
         flows=flows,
     )
 
-    display(data)
+    display(
+        simulated=simulated,
+        measured=measured,
+    )
 
 
 if __name__ == "__main__":
diff --git a/finance/visualize.py b/finance/visualize.py
index e6d609a..c7a7c60 100644
--- a/finance/visualize.py
+++ b/finance/visualize.py
@@ -4,13 +4,24 @@ from decimal import Decimal
 import matplotlib.pyplot
 
 
-def display(data: tuple[list[datetime], list[Decimal]]) -> None:
+def display(
+    simulated: tuple[list[datetime], list[Decimal]],
+    measured: list[tuple[datetime, Decimal]],
+) -> None:
     matplotlib.pyplot.plot(
-        list(data[0]),  # type: ignore
-        [float(i) for i in data[1]],
-        label="Money",
+        list(simulated[0]),  # type: ignore
+        [float(i) for i in simulated[1]],
+        label="Simulated",
+    )
+    matplotlib.pyplot.plot(
+        [i[0] for i in measured],  # type: ignore
+        [float(i[1]) for i in measured],
+        label="Measured",
+        marker="x",
+        linestyle="None",
     )
     matplotlib.pyplot.xlabel("Time")
+    matplotlib.pyplot.ylabel("Money")
     matplotlib.pyplot.legend()
     matplotlib.pyplot.xticks(rotation=45)
     matplotlib.pyplot.show()