Testing: Tests for horizontal and vertical plotting

hyperspy/drawing/utils.py

@@ -164,6 +164,7 @@ def create_figure(window_title=None,
     if "autoscale" in kwargs:
         raise ValueError
     if widget:
+        from IPython.display import display
         with widget:
             fig = plt.figure(**kwargs)
             display(fig.canvas)

hyperspy/tests/drawing/test_horizontal_plotting.py

@@ -0,0 +1,75 @@
+# Copyright 2007-2022 The HyperSpy developers
+#
+# This file is part of HyperSpy.
+#
+# HyperSpy is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# HyperSpy is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with HyperSpy. If not, see <https://www.gnu.org/licenses/#GPL>.
+import pytest
+
+import hyperspy.api as hs
+import numpy as np
+import matplotlib
+from contextlib import redirect_stdout
+import io
+ipympl = pytest.importorskip("ipympl")
+ipywidgets = pytest.importorskip("ipywidgets")
+from ipywidgets import Output
+
+class TestIPYMPL:
+    def test_horizontal(self):
+        matplotlib.use("module://ipympl.backend_nbagg")
+        f = io.StringIO()
+        with redirect_stdout(f):
+            s = hs.signals.Signal2D(np.random.random((4, 4, 2, 2)))
+            s.plot(plot_style="horizontal")
+
+        assert("HBox(children=(Output(layout=Layout(margin='auto 0px auto 0px')),"
+               " Output(layout=Layout(margin='auto 0px auto 0px')" in f.getvalue())
+
+    def test_vertical(self):
+        matplotlib.use("module://ipympl.backend_nbagg")
+        f = io.StringIO()
+        with redirect_stdout(f):
+            s = hs.signals.Signal2D(np.random.random((4, 4, 2, 2)))
+            s.plot(plot_style="vertical")
+
+        assert("VBox(children=(Output(layout=Layout(margin='0px auto 0px auto')),"
+               " Output(layout=Layout(margin='0px auto 0px auto')" in f.getvalue())
+
+    def test_one_output(self):
+        matplotlib.use("module://ipympl.backend_nbagg")
+        f = io.StringIO()
+        with redirect_stdout(f):
+            o = Output()
+            s = hs.signals.Signal2D(np.random.random((4, 4, 2, 2)))
+            s.plot(widget=o)
+        assert("Output()" in f.getvalue())
+
+        with redirect_stdout(f):
+            o = Output()
+            s = hs.signals.Signal2D(np.random.random((4, 4, 2, 2)))
+            s.plot(navigator_kwds={"widget":o})
+        assert("Output()" in f.getvalue())
+
+
+    def test_two_outputs(self):
+        matplotlib.use("module://ipympl.backend_nbagg")
+        f = io.StringIO()
+        with redirect_stdout(f):
+            o = Output()
+            o2 = Output()
+            s = hs.signals.Signal2D(np.random.random((4, 4, 2, 2)))
+            s.plot(widget=o, navigator_kwds={"widget":o2})
+        assert("Output" not in f.getvalue())
+
+