def test_ezqkdef_file_ezgprm_ezgfstp(self):
     rmn.fstopt(rmn.FSTOP_MSGLVL,rmn.FSTOPI_MSG_CATAST)
     ATM_MODEL_DFILES = os.getenv('ATM_MODEL_DFILES')
     myfile = os.path.join(ATM_MODEL_DFILES.strip(),'bcmk/geophy.fst')
     funit = rmn.fstopenall(myfile,rmn.FST_RO)
     (ni,nj) = (201,100)
     gp = {
         'shape' : (ni,nj),
         'ni' : ni,
         'nj' : nj,
         'grtyp' : 'Z',
         'ig1'   : 2002,
         'ig2'   : 1000,
         'ig3'   : 0,
         'ig4'   : 0,
         'grref' : 'E',
         'ig1ref' : 900,
         'ig2ref' : 0,
         'ig3ref' : 43200,
         'ig4ref' : 43200,
         'iunit'  : funit
         }
     gid1 = rmn.ezqkdef(gp)
     a = rmn.ezgfstp(gid1)
     rmn.fstcloseall(funit)
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     self.assertEqual(a['nomvarx'].strip(),'>>')
     self.assertEqual(a['nomvary'].strip(),'^^')
     rmn.gdrls(gid1)
 def test_ezqkdef_file_ezgprm_ezgfstp(self):
     rmn.fstopt(rmn.FSTOP_MSGLVL,rmn.FSTOPI_MSG_CATAST)
     ATM_MODEL_DFILES = os.getenv('ATM_MODEL_DFILES')
     myfile = os.path.join(ATM_MODEL_DFILES.strip(),'bcmk/geophy.fst')
     funit = rmn.fstopenall(myfile,rmn.FST_RO)
     (ni,nj) = (201,100)
     gp = {
         'shape' : (ni,nj),
         'ni' : ni,
         'nj' : nj,
         'grtyp' : 'Z',
         'ig1'   : 2002,
         'ig2'   : 1000,
         'ig3'   : 0,
         'ig4'   : 0,
         'grref' : 'E',
         'ig1ref' : 900,
         'ig2ref' : 0,
         'ig3ref' : 43200,
         'ig4ref' : 43200,
         'iunit'  : funit
         }
     gid1 = rmn.ezqkdef(gp)
     a = rmn.ezgfstp(gid1)
     rmn.fstcloseall(funit)
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     self.assertEqual(a['nomvarx'].strip(),'>>')
     self.assertEqual(a['nomvary'].strip(),'^^')
     rmn.gdrls(gid1)
 def test_ezqkdef_ezgprm(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     rmn.gdrls(gid1)
 def test_ezqkdef_ezgprm(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     rmn.gdrls(gid1)
 def test_ezqkdef_1subgrid(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     ng = rmn.ezget_nsubgrids(gid1)
     self.assertEqual(ng,1)
     subgid = rmn.ezget_subgridids(gid1)
     self.assertEqual(subgid,[gid1])
     rmn.gdrls(gid1)
 def test_ezqkdef_1subgrid(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     ng = rmn.ezget_nsubgrids(gid1)
     self.assertEqual(ng,1)
     subgid = rmn.ezget_subgridids(gid1)
     self.assertEqual(subgid,[gid1])
     rmn.gdrls(gid1)
 def test_ezqkdef_gdll(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     gll = rmn.gdll(gid1)
     self.assertEqual(gp['shape'],gll['lat'].shape)
     self.assertEqual(gp['shape'],gll['lon'].shape)
     self.assertEqual(gp['lat0'],gll['lat'][0,0])
     self.assertEqual(gp['lon0'],gll['lon'][0,0])
     rmn.gdrls(gid1)
 def test_ezqkdef_gdll(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp)
     self.assertTrue(gid1>=0)
     gll = rmn.gdll(gid1)
     self.assertEqual(gp['shape'],gll['lat'].shape)
     self.assertEqual(gp['shape'],gll['lon'].shape)
     self.assertEqual(gp['lat0'],gll['lat'][0,0])
     self.assertEqual(gp['lon0'],gll['lon'][0,0])
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_ezgxprm(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k],'(%s) Expected: %s, Got: %s :: %s' % (k,repr(gp[k]),repr(gprm[k]),repr(gprm)))
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_ezgxprm(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k],'(%s) Expected: %s, Got: %s :: %s' % (k,repr(gp[k]),repr(gprm[k]),repr(gprm)))
     rmn.gdrls(gid1)
 def test_gdxysval(self):
     gp1 = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp1)
     self.assertTrue(gid1>=0)
     zin = np.empty(gp1['shape'],dtype=np.float32,order='FORTRAN')
     for x in range(gp1['ni']):
         zin[:,x] = x
     xx = np.array([1.5],dtype=np.float32,order='FORTRAN')
     yy = np.array([1.5],dtype=np.float32,order='FORTRAN')
     zout = rmn.gdxysval(gid1,xx,yy,zin)
     self.assertEqual(xx.shape,zout.shape)
     self.assertTrue(abs((zin[0,0]+zin[1,1])/2. - zout[0]) < self.epsilon)
     rmn.gdrls(gid1)
 def test_gdllsval(self):
     gp1 = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp1)
     self.assertTrue(gid1>=0)
     zin = np.empty(gp1['shape'],dtype=np.float32,order='FORTRAN')
     for x in range(gp1['ni']):
         zin[:,x] = x
     lat = np.array([gp1['lat0']+gp1['dlat']/2.],dtype=np.float32,order='FORTRAN')
     lon = np.array([(gp1['lon0']+gp1['dlon'])/2.],dtype=np.float32,order='FORTRAN')
     zout = rmn.gdllsval(gid1,lat,lon,zin)
     self.assertEqual(lat.shape,zout.shape)
     self.assertTrue(abs((zin[0,0]+zin[1,1])/2. - zout[0]) < self.epsilon)
     rmn.gdrls(gid1)
 def test_gdllsval(self):
     gp1 = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp1)
     self.assertTrue(gid1>=0)
     zin = np.empty(gp1['shape'],dtype=np.float32,order='FORTRAN')
     for x in xrange(gp1['ni']):
         zin[:,x] = x
     lat = np.array([gp1['lat0']+gp1['dlat']/2.],dtype=np.float32,order='FORTRAN')
     lon = np.array([(gp1['lon0']+gp1['dlon'])/2.],dtype=np.float32,order='FORTRAN')
     zout = rmn.gdllsval(gid1,lat,lon,zin)
     self.assertEqual(lat.shape,zout.shape)
     self.assertTrue(abs((zin[0,0]+zin[1,1])/2. - zout[0]) < self.epsilon)
     rmn.gdrls(gid1)
 def test_gdxysval(self):
     gp1 = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp1)
     self.assertTrue(gid1>=0)
     zin = np.empty(gp1['shape'],dtype=np.float32,order='FORTRAN')
     for x in xrange(gp1['ni']):
         zin[:,x] = x
     xx = np.array([1.5],dtype=np.float32,order='FORTRAN')
     yy = np.array([1.5],dtype=np.float32,order='FORTRAN')
     zout = rmn.gdxysval(gid1,xx,yy,zin)
     self.assertEqual(xx.shape,zout.shape)
     self.assertTrue(abs((zin[0,0]+zin[1,1])/2. - zout[0]) < self.epsilon)
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdgaxes(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     axes = rmn.gdgaxes(gid1)
     self.assertEqual(axes['ax'].shape,gp['ax'].shape)
     self.assertEqual(axes['ay'].shape,gp['ay'].shape)
     for i in xrange(gp['ni']):
         self.assertTrue(abs(axes['ax'][i,0]-gp['ax'][i,0])<self.epsilon)
     for j in xrange(gp['nj']):
         self.assertTrue(abs(axes['ay'][0,j]-gp['ay'][0,j])<self.epsilon)
     rmn.gdrls(gid1)
 def test_ezqkdef_ezgxprm(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp['ni'],gp['nj'],gp['grtyp'],
                        gp['ig1'],gp['ig2'],gp['ig3'],gp['ig4'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gp['grref'] = ''
     gp['ig1ref'] = 0
     gp['ig2ref'] = 0
     gp['ig3ref'] = 0
     gp['ig4ref'] = 0
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     rmn.gdrls(gid1)
 def test_ezqkdef_ezgxprm(self):
     gp = self.getGridParams_L()
     gid1 = rmn.ezqkdef(gp['ni'],gp['nj'],gp['grtyp'],
                        gp['ig1'],gp['ig2'],gp['ig3'],gp['ig4'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     gp['grref'] = ''
     gp['ig1ref'] = 0
     gp['ig2ref'] = 0
     gp['ig3ref'] = 0
     gp['ig4ref'] = 0
     gprm = rmn.ezgxprm(gid1)
     for k in gprm.keys():
         self.assertEqual(gp[k],gprm[k])
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdgaxes(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     gp['id'] = gid1
     axes = rmn.gdgaxes(gid1)
     self.assertEqual(axes['ax'].shape,gp['ax'].shape)
     self.assertEqual(axes['ay'].shape,gp['ay'].shape)
     for i in range(gp['ni']):
         self.assertTrue(abs(axes['ax'][i,0]-gp['ax'][i,0])<self.epsilon)
     for j in range(gp['nj']):
         self.assertTrue(abs(axes['ay'][0,j]-gp['ay'][0,j])<self.epsilon)
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdxyfll(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     lat = np.array([gp['ay'][0,0],gp['ay'][0,1]],dtype=np.float32,order='FORTRAN')
     lon = np.array([gp['ax'][0,0],gp['ax'][1,0]],dtype=np.float32,order='FORTRAN')
     xypts = rmn.gdxyfll(gid1, lat, lon)
     self.assertEqual(xypts['x'].shape,lat.shape)
     self.assertEqual(xypts['y'].shape,lat.shape)
     self.assertTrue(abs(xypts['x'][0]-1.)<self.epsilon)
     self.assertTrue(abs(xypts['y'][0]-1.)<self.epsilon)
     self.assertTrue(abs(xypts['x'][1]-2.)<self.epsilon)
     self.assertTrue(abs(xypts['y'][1]-2.)<self.epsilon)
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdxyfll(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     lat = np.array([gp['ay'][0,0],gp['ay'][0,1]],dtype=np.float32,order='FORTRAN')
     lon = np.array([gp['ax'][0,0],gp['ax'][1,0]],dtype=np.float32,order='FORTRAN')
     xypts = rmn.gdxyfll(gid1, lat, lon)
     self.assertEqual(xypts['x'].shape,lat.shape)
     self.assertEqual(xypts['y'].shape,lat.shape)
     self.assertTrue(abs(xypts['x'][0]-1.)<self.epsilon)
     self.assertTrue(abs(xypts['y'][0]-1.)<self.epsilon)
     self.assertTrue(abs(xypts['x'][1]-2.)<self.epsilon)
     self.assertTrue(abs(xypts['y'][1]-2.)<self.epsilon)
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdllfxy(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     xx = np.array([1.,2.],dtype=np.float32,order='FORTRAN')
     yy = np.array([1.,3.],dtype=np.float32,order='FORTRAN')
     llpts = rmn.gdllfxy(gid1, xx, yy)
     self.assertEqual(llpts['x'].shape,xx.shape)
     self.assertEqual(llpts['y'].shape,xx.shape)
     self.assertTrue(abs(llpts['lon'][0]-gp['ax'][0,0])<self.epsilon)
     self.assertTrue(abs(llpts['lat'][0]-gp['ay'][0,0])<self.epsilon)
     self.assertTrue(abs(llpts['lon'][1]-gp['ax'][1,0])<self.epsilon)
     self.assertTrue(abs(llpts['lat'][1]-gp['ay'][0,2])<self.epsilon)
     rmn.gdrls(gid1)
 def test_ezgkdef_fmem_gdllfxy(self):
     gp = self.getGridParams_ZE()
     gid1 = rmn.ezgdef_fmem(gp['ni'],gp['nj'],gp['grtyp'],gp['grref'],
                            gp['ig1ref'],gp['ig2ref'],gp['ig3ref'],gp['ig4ref'],
                            gp['ax'],gp['ay'])
     self.assertTrue(gid1>=0)
     xx = np.array([1.,2.],dtype=np.float32,order='FORTRAN')
     yy = np.array([1.,3.],dtype=np.float32,order='FORTRAN')
     llpts = rmn.gdllfxy(gid1, xx, yy)
     self.assertEqual(llpts['x'].shape,xx.shape)
     self.assertEqual(llpts['y'].shape,xx.shape)
     self.assertTrue(abs(llpts['lon'][0]-gp['ax'][0,0])<self.epsilon)
     self.assertTrue(abs(llpts['lat'][0]-gp['ay'][0,0])<self.epsilon)
     self.assertTrue(abs(llpts['lon'][1]-gp['ax'][1,0])<self.epsilon)
     self.assertTrue(abs(llpts['lat'][1]-gp['ay'][0,2])<self.epsilon)
     rmn.gdrls(gid1)