def testlonlat2cell_hist(self): """ setup grid with unequal cell size along lat and lon and test if the correct number of points lay in each cell """ cells = lonlat2cell( self.lons, self.lats, cellsize_lon=15, cellsize_lat=30) hist, bin_edges = np.histogram( cells.flatten(), bins=len(np.unique(cells))) nptest.assert_allclose(hist, np.zeros_like(hist) + 72)
def setUp(self): lat, lon = np.arange(180) - 90, np.arange(360) - 180 self.lats, self.lons = np.meshgrid(lat, lon) self.lats, self.lons = self.lats.flatten(), self.lons.flatten() self.cells = grids.lonlat2cell(self.lons, self.lats) self.subset = np.sort(np.random.choice(np.arange(self.lats.size), size=500, replace=False)) self.basic = grids.BasicGrid(self.lons, self.lats, subset=self.subset, shape=(360, 180)) self.basic_irregular = grids.BasicGrid(self.lons, self.lats, subset=self.subset) self.cellgrid = grids.CellGrid(self.lons, self.lats, self.cells, subset=self.subset) self.testfilename = os.path.join(curpath(), "data", "test.nc") if not os.path.exists(os.path.join(curpath(), "data")): os.mkdir(os.path.join(curpath(), "data"))