TST: add a test

jklymak · jklymak · commit a66c671e32f2 · 2020-12-09T22:30:45.000-08:00
diff --git a/lib/matplotlib/tests/test_scale.py b/lib/matplotlib/tests/test_scale.py
@@ -2,6 +2,7 @@
 from matplotlib.scale import (
     LogTransform, InvertedLogTransform,
     SymmetricalLogTransform)
+import matplotlib.scale as mscale
 from matplotlib.testing.decorators import check_figures_equal, image_comparison
 
 import numpy as np
@@ -198,3 +199,14 @@ def forward(x):
     ax.plot(x, x)
     ax.set_xscale('function', functions=(forward, inverse))
     ax.set_xlim(1, 1000)
+
+
+def test_pass_scale():
+    # test passing a scale object works...
+    fig, ax = plt.subplots()
+    scale = mscale.LogScale(axis=None)
+    ax.set_xscale(scale)
+    scale = mscale.LogScale(axis=None)
+    ax.set_yscale(scale)
+    assert ax.xaxis.get_scale() == 'log'
+    assert ax.yaxis.get_scale() == 'log'
