Move display() to finance.visualize.py

2 files changed, 16 insertions, 14 deletions

diff --git a/finance/flow.py b/finance/flow.py
index bc2ee0a..4cf68db 100644
--- a/finance/flow.py
+++ b/finance/flow.py
@@ -3,8 +3,6 @@ from datetime import datetime
 from decimal import Decimal
 from typing import Generator
 
-import matplotlib.pyplot
-
 
 @dataclasses.dataclass(kw_only=True, frozen=True)
 class Flow:
@@ -66,15 +64,3 @@ def simulate(
         values.append(value)
 
     return (dates, values)
-
-
-def display(data: tuple[list[datetime], list[Decimal]]) -> None:
-    matplotlib.pyplot.plot(
-        list(data[0]),  # type: ignore
-        [float(i) for i in data[1]],
-        label="Money",
-    )
-    matplotlib.pyplot.xlabel("Time")
-    matplotlib.pyplot.legend()
-    matplotlib.pyplot.xticks(rotation=45)
-    matplotlib.pyplot.show()
diff --git a/finance/visualize.py b/finance/visualize.py
new file mode 100644
index 0000000..e6d609a
--- /dev/null
+++ b/finance/visualize.py
@@ -0,0 +1,16 @@
+from datetime import datetime
+from decimal import Decimal
+
+import matplotlib.pyplot
+
+
+def display(data: tuple[list[datetime], list[Decimal]]) -> None:
+    matplotlib.pyplot.plot(
+        list(data[0]),  # type: ignore
+        [float(i) for i in data[1]],
+        label="Money",
+    )
+    matplotlib.pyplot.xlabel("Time")
+    matplotlib.pyplot.legend()
+    matplotlib.pyplot.xticks(rotation=45)
+    matplotlib.pyplot.show()