def setUp(self):
        roms = 'ocean_s_coordinate_g1_roms.nc'
        self.nc = netCDF4.Dataset(os.path.join(data_path, roms))
        formula_terms_variable = get_formula_terms_variables(self.nc)[0]
        formula_terms = get_formula_terms(formula_terms_variable)
        formula_terms_arrays = nc2biggus(self.nc, formula_terms)

        s = formula_terms_arrays['s']
        c = formula_terms_arrays['C']
        eta = formula_terms_arrays['eta']
        depth = formula_terms_arrays['depth']
        depth_c = formula_terms_arrays['depth_c']

        self.z = ocean_s_coordinate_g1(s, c, eta, depth, depth_c)
        self.sliced = self.z[0, :, 30, 80]
Example #2
0
 def test_formula_terms_variables(self):
     formula_terms_variables = get_formula_terms_variables(self.nc)
     for var in formula_terms_variables:
         assert hasattr(var,  'formula_terms')
Example #3
0
 def test_no_formula_terms_variables(self):
     no_formula_term = 'no_formula_terms.nc'
     with netCDF4.Dataset(os.path.join(data_path, no_formula_term)) as nc:
         with pytest.raises(ValueError):
             get_formula_terms_variables(nc)
Example #4
0
 def setUp(self):
     roms = 'ocean_s_coordinate_g1_roms.nc'
     self.nc = netCDF4.Dataset(os.path.join(data_path, roms))
     self.formula_terms_variable = get_formula_terms_variables(self.nc)[0]
     self.formula_terms = get_formula_terms(self.formula_terms_variable)
     self.formula_terms_arrays = nc2biggus(self.nc, self.formula_terms)