Rewrite an argument check to _check_getitem

timhoffm · timhoffm · commit 0e29b47b1e94 · 2019-07-02T07:36:49.000+02:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2858,14 +2858,8 @@ def ticklabel_format(self, *, axis='both', style='', scilimits=None,
                 m + n + 1  # check that both are numbers
             except (ValueError, TypeError):
                 raise ValueError("scilimits must be a sequence of 2 integers")
-        if style[:3] == 'sci':
-            sb = True
-        elif style == 'plain':
-            sb = False
-        elif style == '':
-            sb = None
-        else:
-            raise ValueError("%s is not a valid style value")
+        STYLES = {'sci': True, 'scientific': True, 'plain': False, '': None}
+        sb = cbook._check_getitem(STYLES, style=style)
         try:
             if sb is not None:
                 if axis == 'both' or axis == 'x':
