fix: simplify

andrzejnovak · andrzejnovak · commit 3253a3e6eebd · 2020-03-02T18:11:57.000+01:00
diff --git a/lib/matplotlib/mathtext.py b/lib/matplotlib/mathtext.py
@@ -768,22 +768,17 @@ class UnicodeFonts(TruetypeFonts):
     use_cmex = True
 
     def __init__(self, *args, **kwargs):
-        # This must come first so the backend's owner is set correctly
         fallback_rc = rcParams['mathtext.fallback']
         if rcParams['mathtext.fallback_to_cm'] is not None:
-            if rcParams['mathtext.fallback_to_cm']:
-                fallback_rc = 'cm'
-            else:
-                fallback_rc = None
+            fallback_rc = ('cm' if rcParams['mathtext.fallback_to_cm']
+                           else None)
+
+        font_class = {'stix': StixFonts,
+                      'stixsans': StixSansFonts,
+                      'cm': BakomaFonts
+                      }.get(fallback_rc)
+        self.cm_fallback = font_class(*args, **kwargs) if font_class else None
 
-        if fallback_rc == 'stix':
-            self.cm_fallback = StixFonts(*args, **kwargs)
-        elif fallback_rc == 'stixsans':
-            self.cm_fallback = StixSansFonts(*args, **kwargs)
-        elif fallback_rc == 'cm':
-            self.cm_fallback = BakomaFonts(*args, **kwargs)
-        else:
-            self.cm_fallback = None
         TruetypeFonts.__init__(self, *args, **kwargs)
         self.fontmap = {}
         for texfont in "cal rm tt it bf sf".split():
