Fix very-edge case in csd(), plus small additional cleanups.

anntzer · anntzer · commit 53edb190fd3c · 2021-09-27T12:48:25.000+02:00
Previously, `csd([0, 0], [0, 0])` would fail with a ZeroDivisionError
when computing yticks.  Fix that by copying the logic from psd, and
streamlining it in both places.  Also fix an unformatted exception
message in specgram.
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -7035,12 +7035,11 @@ def psd(self, x, NFFT=None, Fs=None, Fc=None, detrend=None,
         self.set_xlabel('Frequency')
         self.set_ylabel('Power Spectral Density (%s)' % psd_units)
         self.grid(True)
+
         vmin, vmax = self.viewLim.intervaly
         intv = vmax - vmin
         logi = int(np.log10(intv))
-        if logi == 0:
-            logi = .1
-        step = 10 * logi
+        step = max(10 * int(np.log10(intv)), 1)
         ticks = np.arange(math.floor(vmin), math.ceil(vmax) + 1, step)
         self.set_yticks(ticks)
 
@@ -7140,11 +7139,10 @@ def csd(self, x, y, NFFT=None, Fs=None, Fc=None, detrend=None,
         self.set_xlabel('Frequency')
         self.set_ylabel('Cross Spectrum Magnitude (dB)')
         self.grid(True)
-        vmin, vmax = self.viewLim.intervaly
 
+        vmin, vmax = self.viewLim.intervaly
         intv = vmax - vmin
-        step = 10 * int(np.log10(intv))
-
+        step = max(10 * int(np.log10(intv)), 1)
         ticks = np.arange(math.floor(vmin), math.ceil(vmax) + 1, step)
         self.set_yticks(ticks)
 
@@ -7576,7 +7574,7 @@ def specgram(self, x, NFFT=None, Fs=None, Fc=None, detrend=None,
             else:
                 Z = 20. * np.log10(spec)
         else:
-            raise ValueError('Unknown scale %s', scale)
+            raise ValueError('Unknown scale {scale!r}')
 
         Z = np.flipud(Z)
 
