converted assert into exception

montefra · montefra · commit 873e91d6c9ef · 2015-03-04T23:48:30.000+01:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -50,6 +50,8 @@
 
 docstring.interpd.update(projection_names=get_projection_names())
 
+class AxesException(Exception):
+    pass
 
 class AxesStack(Stack):
     """
@@ -872,7 +874,9 @@ def add_axes(self, *args, **kwargs):
 
         if isinstance(args[0], Axes):
             a = args[0]
-            assert(a.get_figure() is self)
+            if a.get_figure() is not self:
+                msg = "The Axes must have been created in the present figure"
+                raise AxesException(msg)
         else:
             rect = args[0]
             projection_class, kwargs, key = process_projection_requirements(
@@ -946,7 +950,9 @@ def add_subplot(self, *args, **kwargs):
         if isinstance(args[0], SubplotBase):
 
             a = args[0]
-            assert(a.get_figure() is self)
+            if a.get_figure() is not self:
+                msg = "The Subplot must have been created in the present figure"
+                raise AxesException(msg)
             # make a key for the subplot (which includes the axes object id
             # in the hash)
             key = self._make_key(*args, **kwargs)
