Fix LaTeX error due to labels that become empty in simplify_labels

Fixes astropy#8004.
lpsinger · Mar 12, 2020 · f0deda9 · f0deda9
1 parent 034fc83
commit f0deda9
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -488,6 +488,9 @@ astropy.visualization
   maximum when there are less then ``min_npixels`` in the input array.
   [#9913]
 
+- Fix a bug in simplifying axis labels that affected non-rectangular frames.
+  [#8004, #9991]
+
 astropy.wcs
 ^^^^^^^^^^^
 

diff --git a/astropy/visualization/wcsaxes/tests/test_misc.py b/astropy/visualization/wcsaxes/tests/test_misc.py
@@ -18,12 +18,15 @@
 from astropy.tests.image_tests import ignore_matplotlibrc
 
 from astropy.visualization.wcsaxes.core import WCSAxes
-from astropy.visualization.wcsaxes.frame import RectangularFrame, RectangularFrame1D
+from astropy.visualization.wcsaxes.frame import (
+    EllipticalFrame, RectangularFrame, RectangularFrame1D)
 from astropy.visualization.wcsaxes.utils import get_coord_meta
 from astropy.visualization.wcsaxes.transforms import CurvedTransform
 
 MATPLOTLIB_LT_21 = LooseVersion(matplotlib.__version__) < LooseVersion("2.1")
 
+TEX_MISSING = not matplotlib.checkdep_usetex(True)
+
 DATA = os.path.abspath(os.path.join(os.path.dirname(__file__), 'data'))
 
 
@@ -446,3 +449,37 @@ def test_time_wcs(time_spectral_wcs_2d):
     # with a time axis.
 
     plt.subplot(projection=time_spectral_wcs_2d)
+
+
+@pytest.mark.skipif('TEX_MISSING')
+@pytest.mark.skipif('MATPLOTLIB_LT_21')
+@ignore_matplotlibrc
+def test_simplify_labels_usetex(tmpdir):
+    """Regression test for https://github.com/astropy/astropy/issues/8004."""
+    plt.rc('text', usetex=True)
+
+    header = {
+        'NAXIS': 2,
+        'NAXIS1': 360,
+        'NAXIS2': 180,
+        'CRPIX1': 180.5,
+        'CRPIX2': 90.5,
+        'CRVAL1': 180.0,
+        'CRVAL2': 0.0,
+        'CDELT1': -2 * np.sqrt(2) / np.pi,
+        'CDELT2': 2 * np.sqrt(2) / np.pi,
+        'CTYPE1': 'RA---MOL',
+        'CTYPE2': 'DEC--MOL',
+        'RADESYS': 'ICRS'}
+
+    wcs = WCS(header)
+    fig, ax = plt.subplots(
+        subplot_kw=dict(frame_class=EllipticalFrame, projection=wcs))
+    ax.set_xlim(-0.5, header['NAXIS1'] - 0.5)
+    ax.set_ylim(-0.5, header['NAXIS2'] - 0.5)
+    ax.coords[0].set_ticklabel(exclude_overlapping=True)
+    ax.coords[1].set_ticklabel(exclude_overlapping=True)
+    ax.coords[0].set_ticks(spacing=45 * u.deg)
+    ax.coords[1].set_ticks(spacing=30 * u.deg)
+    ax.grid()
+    fig.savefig(tmpdir / 'plot.png')
diff --git a/astropy/visualization/wcsaxes/ticklabels.py b/astropy/visualization/wcsaxes/ticklabels.py
@@ -93,6 +93,8 @@ def simplify_labels(self):
                     self.text[axis][i] = self.text[axis][i][start:]
                     if starts_dollar:
                         self.text[axis][i] = '$' + self.text[axis][i]
+                if self.text[axis][i] == '$$':
+                    self.text[axis][i] = ''
 
     def set_pad(self, value):
         self._pad = value