Пример #1
0
 def test(self):
     self.assertEqual(namelist([{'name': 'Bart'},{'name': 'Lisa'},{'name': 'Maggie'},{'name': 'Homer'},{'name': 'Marge'}]), 'Bart, Lisa, Maggie, Homer & Marge',
 "Must work with many names")
     self.assertEqual(namelist([{'name': 'Bart'},{'name': 'Lisa'},{'name': 'Maggie'}]), 'Bart, Lisa & Maggie',
 "Must work with many names")
     self.assertEqual(namelist([{'name': 'Bart'},{'name': 'Lisa'}]), 'Bart & Lisa', 
 "Must work with two names")
     self.assertEqual(namelist([{'name': 'Bart'}]), 'Bart', "Wrong output for a single name")
     self.assertEqual(namelist([]), '', "Must work with no names")
Пример #2
0
    def test_many_names(self):
        names = [{
            'name': 'Bart'
        }, {
            'name': 'Lisa'
        }, {
            'name': 'Maggie'
        }, {
            'name': 'Homer'
        }, {
            'name': 'Marge'
        }]
        self.assertEqual(namelist(names), 'Bart, Lisa, Maggie, Homer & Marge')

        names = [{'name': 'Bart'}, {'name': 'Lisa'}, {'name': 'Maggie'}]
        self.assertEqual(namelist(names), 'Bart, Lisa & Maggie')
Пример #3
0
 def test_rand(self):
     def solist(names): return (lambda z: ", ".join(z[:-1])+" & "+z[-1] if len(z)>1 else "".join(z))((lambda y: [x['name'] for x in y])(names))
     base=[{'name': 'Bart'},{'name': 'Lisa'},{'name': 'Maggie'},{'name': 'Homer'},{'name': 'Marge'},{'name': 'Moe'},{'name': 'Barney'},{'name': 'Maude'},{'name': 'Ned'},{'name': 'Seymour'}]
     for _ in range(40):
         shuffle(base)
         testmat=base[:randint(0,9)]
         solution=solist(testmat)
         self.assertEqual(namelist(deepcopy(testmat)),solution,"It should work for random tests too")
Пример #4
0
 def test_3(self):
     result = namelist([{
         "name": "Bart"
     }, {
         "name": "Lisa"
     }, {
         "name": "Maggie"
     }])
     self.assertEqual(result, "Bart, Lisa & Maggie")
Пример #5
0
def setvertlevs(vfile, orog):

    txt = open(vfile).read()
    vertlevs = namelist.namelist(txt,1)
    # Only a single namelist, second elment is the dictionary
    vertlevs = vertlevs.next()[1]
    print "VERTLEVS", vertlevs

    eta_theta_levels  = np.array(vertlevs['eta_theta']) # (0:model_levels)
    eta_rho_levels = np.array([-1e20] + vertlevs['eta_rho'])  # (1:model_levels)
    z_top_of_model = vertlevs['z_top_of_model']
    first_constant_r_rho_level = vertlevs['first_constant_r_rho_level']
    print eta_rho_levels

    ashape = (len(eta_theta_levels),) + orog.shape
    print "ASHAPE", ashape
    
    r_theta_levels = np.zeros(ashape)
    r_rho_levels = np.zeros(ashape)

    r_ref_theta = eta_theta_levels * z_top_of_model
    r_ref_rho = eta_rho_levels * z_top_of_model
    #  set bottom level, ie: orography
    r_theta_levels[0] = orog[:] # + Earth_radius

    #  For constant levels set r to be a constant on the level
    if len(orog.shape) == 2:
        r_theta_levels[first_constant_r_rho_level:] = r_ref_theta[first_constant_r_rho_level:,np.newaxis,np.newaxis]
        r_rho_levels[first_constant_r_rho_level:] = r_ref_rho[first_constant_r_rho_level:,np.newaxis,np.newaxis]
    else:
        r_theta_levels[first_constant_r_rho_level:] = r_ref_theta[first_constant_r_rho_level:,np.newaxis]
        r_rho_levels[first_constant_r_rho_level:] = r_ref_rho[first_constant_r_rho_level:,np.newaxis]

 
    #  Case( height_gen_smooth )
    # A smooth quadratic height generation
    for k in range(1, first_constant_r_rho_level):
        r_rho_levels[k] = eta_rho_levels[k] * z_top_of_model + \
          orog * (1.0 - eta_rho_levels[k]/eta_rho_levels[first_constant_r_rho_level])**2
        r_theta_levels[k] = eta_theta_levels[k] * z_top_of_model + \
          orog * (1.0 - eta_theta_levels[k]/eta_rho_levels[first_constant_r_rho_level])**2

    return r_theta_levels, r_rho_levels
Пример #6
0
import multigrid as mgmpi  #routines for the parallel multigrid pressure solver
import explicit_schemes as expl  #routines to calculate all the explicit tendendies in the Navier-Stokes equations
from rk_tables import rk_coef  #coefficients table for the explicit time scheme
import output_fields as out  #routines to deal with model output
import namelist  #read in the pre-edited namelist file (./namelist.py)
from init_grid import init_grid  #read in the grid information
from init_emiss import get_emissions  #read in the emission file (optionally)
import coupling as cpl  #routines for mesoscale coupling and lateral boundary conditions

#MPI communicator
mpicomm = MPI.COMM_WORLD
nproc = mpicomm.Get_size()
rank = mpicomm.Get_rank()

#read namelist (default path is ./namelist)
param_dict = namelist.namelist()
if rank == 0:
    print "SIMULATION SETTINGS:"
    print ""
    for param in param_dict:
        print param + ': ' + str(param_dict[param])
    print ""
    print "Simulation initializing..."

npc = int(param_dict['npc'])
npr = int(param_dict['npr'])

ng = int(param_dict['n_ghost'])
ng1 = ng - 1

if nproc != npc * npr:
Пример #7
0
 def test_no_names(self):
     names = []
     self.assertEqual(namelist(names), '')
Пример #8
0
 def test_one_name(self):
     names = [{'name': 'Bart'}]
     self.assertEqual(namelist(names), 'Bart')
Пример #9
0
 def test_two_names(self):
     names = [{'name': 'Bart'}, {'name': 'Lisa'}]
     self.assertEqual(namelist(names), 'Bart & Lisa')
Пример #10
0
try:
    if (sys.argv[1] == "-nc"):
        filetype = "netcdf"
    elif (sys.argv[1] == "-grib"):
        filetype = "grib"
    else:
        print "Not valid filetype, use -h or --help to get help"
except IndexError:
    pass

if filetype == "netcdf":
    print "nothing here yet"

if filetype == "grib":
    data_path, fileprefix, filepostfix, startdate, enddate, timestep = namelist.namelist()


m = Basemap(llcrnrlon=-15,llcrnrlat=45,urcrnrlon=30,urcrnrlat=68,
            resolution='i',projection='stere',
            lat_0=50,lon_0=6)

dl = startdate
indx = 0

locs = []
xlocs = []
ylocs = []
latlocs = []
lonlocs = []
Пример #11
0
 def test_2(self):
     result = namelist([{"name": "Bart"}, {"name": "Lisa"}])
     self.assertEqual(result, "Bart & Lisa")
Пример #12
0
 def test_1(self):
     result = namelist([{"name": "Bart"}])
     self.assertEqual(result, "Bart")