Add show_plot option for Solution.plot()

src/thevenin/_solutions.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 from typing import Iterable, TYPE_CHECKING
 
-import atexit
 import textwrap
 from copy import deepcopy
 
+import atexit
 import numpy as np
 import matplotlib.pyplot as plt
 from scipy.integrate import cumulative_trapezoid
@@ -74,7 +74,7 @@ def wrap_string(label: str, value: list, width: int):
 
         return readable
 
-    def plot(self, x: str, y: str, **kwargs) -> None:
+    def plot(self, x: str, y: str, show_plot: bool = True, **kwargs) -> None:
         """
         Plot any two variables in 'vars' against each other.
 
@@ -84,6 +84,10 @@ def plot(self, x: str, y: str, **kwargs) -> None:
             A variable key in 'vars' to be used for the x-axis.
         y : str
             A variable key in 'vars' to be used for the y-axis.
+        show_plot : bool, optional
+            For non-interactive environments only. When True (default) this
+            registers `plt.show()` to run at the end of the program. If False,
+            you must call `plt.show()` manually.
 
         Returns
         -------
@@ -109,7 +113,8 @@ def plot(self, x: str, y: str, **kwargs) -> None:
         plt.xlabel(xlabel)
         plt.ylabel(ylabel)
 
-        atexit.register(plt.show)
+        if show_plot and not plt.isinteractive():
+            atexit.register(plt.show)
 
     def _to_dict(self) -> None:
         """