SF.net SVN: matplotlib:[5960] trunk/toolkits/basemap/lib/mpl_toolkits/ basemap/__init__.py

[<js...@us...>]

Revision: 5960
          http://matplotlib.svn.sourceforge.net/matplotlib/?rev=5960&view=rev
Author:   jswhit
Date:     2008-08-04 11:49:31 +0000 (Mon, 04 Aug 2008)

Log Message:
-----------
additional shape checking in rotate_vector.

Modified Paths:
--------------
    trunk/toolkits/basemap/lib/mpl_toolkits/basemap/__init__.py

Modified: trunk/toolkits/basemap/lib/mpl_toolkits/basemap/__init__.py
===================================================================
--- trunk/toolkits/basemap/lib/mpl_toolkits/basemap/__init__.py	2008-08-04 11:44:30 UTC (rev 5959)
+++ trunk/toolkits/basemap/lib/mpl_toolkits/basemap/__init__.py	2008-08-04 11:49:31 UTC (rev 5960)
@@ -2423,6 +2423,9 @@
            uin.shape[1] == vin.shape[1] == lons.shape[0] and\
            uin.shape[0] == vin.shape[0] == lats.shape[0]:
             lons, lats = np.meshgrid(lons, lats) 
+        else:
+            if not lons.shape == lats.shape == uin.shape == vin.shape:
+                raise TypeError("shapes of lons,lats and uin,vin don't match")
         x, y = self(lons, lats)
         # rotate from geographic to map coordinates.
         if ma.isMaskedArray(uin):


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.