Clean up location check

dstansby · dstansby · commit 2e4b17e3183f · 2017-01-17T13:18:59.000Z
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -1043,12 +1043,12 @@ def __init__(self, loc,
         self.set_child(child)
 
         if is_string_like(loc):
-            if loc not in self.codes:
+            try:
+                loc = self.codes[loc]
+            except KeyError:
                 raise RuntimeError('Unrecognized location "%s". Valid '
                                    'locations are\n\t%s\n'
                                    % (loc, '\n\t'.join(self.codes)))
-            else:
-                loc = self.codes[loc]
 
         self.loc = loc
         self.borderpad = borderpad
