add plt.close

magicclass/widgets/logger.py

@@ -628,7 +628,7 @@ def set_plt(self, rc_context: dict[str, Any]) -> None:
         ...
 
     @contextmanager
-    def set_plt(self, style: str = None, rc_context: dict[str, Any] = {}):
+    def set_plt(self, style: str | None = None, rc_context: dict[str, Any] = {}):
         """A context manager for inline plot in the logger widget."""
         try:
             import matplotlib as mpl
@@ -657,7 +657,9 @@ def set_plt(self, style: str = None, rc_context: dict[str, Any] = {}):
         backend = mpl.get_backend()
         show._called = False
         try:
-            mpl.use("module://magicclass.widgets.logger")
+            if backend != "module://magicclass.widgets.logger":
+                plt.close("all")
+                mpl.use("module://magicclass.widgets.logger")
             with plt.style.context(style), plt.rc_context(rc_context):
                 yield self
         finally:

magicclass/widgets/tests/test_logger.py

@@ -0,0 +1,14 @@
+from magicclass.widgets import Logger
+
+def test_logger():
+    log = Logger()
+    log.print("Hello")
+    log.print_table({"a": [1, 2], "b": [3, 4]})
+    log.print_table([["a", "b"], [4, -1], [4.3, None]])
+    log.print_html("<b>bold</b>")
+    with log.set_stdout():
+        print("Hello")
+    with log.set_plt():
+        import matplotlib.pyplot as plt
+        plt.plot([1, 2, 3])
+    log.clear()

pyproject.toml

@@ -49,6 +49,7 @@ testing = [
     "numpy",
     "pandas>=1.5.0",
     "polars>=0.17.10",
+    "matplotlib>=3.8.2",
 ]
 docs = [
     "mkdocs",