bug fix: check the type of the 'key' of the two array 'r1' and 'r2'

montefra · tobias47n9e · commit f55f196d2951 · 2014-06-25T18:12:47.000+02:00
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -2631,8 +2631,11 @@ def key_desc(name):
         if dt1.type != np.string_:
             return (name, dt1.descr[0][1])
 
-        dt2 = r1.dtype[name]
-        assert dt2==dt1
+        dt2 = r2.dtype[name]
+        if dt1 != dt2:
+            msg = "The type of the '{}' in the arrays 'r1' and 'r2' must be of"
+            msg += " the same type"
+            raise ValueError(msg.format(name))
         if dt1.num>dt2.num:
             return (name, dt1.descr[0][1])
         else:
