Micro-optimize check_isinstance.

anntzer · anntzer · commit 5f5a31f75e8d · 2020-11-23T01:15:09.000+01:00
- If a single type is passed in, save an iteration over a tuple.
- Don't compute the type names for the error message unless an error
  actually occurred.

Together, this speeds up `_check_isinstance(int, x=1)` by more than
2-fold, and `_check_isinstance((float, int), x=1)` by a few dozen-%.
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -2241,21 +2241,22 @@ def _check_isinstance(_types, **kwargs):
     >>> cbook._check_isinstance((SomeClass, None), arg=arg)
     """
     types = _types
-    if isinstance(types, type) or types is None:
-        types = (types,)
-    none_allowed = None in types
-    types = tuple(tp for tp in types if tp is not None)
+    none_type = type(None)
+    types = ((types,) if isinstance(types, type) else
+             (none_type,) if types is None else
+             tuple(none_type if tp is None else tp for tp in types))
 
     def type_name(tp):
-        return (tp.__qualname__ if tp.__module__ == "builtins"
+        return ("None" if tp is none_type
+                else tp.__qualname__ if tp.__module__ == "builtins"
                 else f"{tp.__module__}.{tp.__qualname__}")
 
-    names = [*map(type_name, types)]
-    if none_allowed:
-        types = (*types, type(None))
-        names.append("None")
     for k, v in kwargs.items():
         if not isinstance(v, types):
+            names = [*map(type_name, types)]
+            if "None" in names:  # Move it to the end for better wording.
+                names.remove("None")
+                names.append("None")
             raise TypeError(
                 "{!r} must be an instance of {}, not a {}".format(
                     k,
