Ejemplo n.º 1
0
 def test_identify_variable(self):
     checklist = [('time', 'time'), ('latitude', 'lat'),
                  ('longitude', 'lon'),
                  ('atmosphere_hybrid_pressure_coordinate', 'hyam'),
                  ('atmosphere_hybrid_height_coordinate', 'hybm'),
                  ('cloud_area_fraction_in_atmosphere_layer',
                   'cloud_area_fraction_in_atmosphere_layer')]
     for standard_name, short_name in checklist:
         variable = identify_variable(self.ncfile_ml, standard_name)
         assert variable[0] == short_name
Ejemplo n.º 2
0
    def _find_data_vars(self):
        """Find NetCDF variables of required data fields.

        A dictionary data_vars is created. Its keys are the CF standard names
        of the variables provided by the plot object. The values are pointers
        to the NetCDF variable objects.

        <data_vars> can be accessed as <self.data_vars>.
        """
        self.data_vars = {}
        self.data_units = {}
        for df_type, df_name, _ in self.plot_object.required_datafields:
            varname, var = netCDF4tools.identify_variable(self.dataset,
                                                          df_name,
                                                          check=True)
            logging.debug("\tidentified variable <%s> for field <%s>", varname,
                          df_name)
            self.data_vars[df_name] = var
            self.data_units[df_name] = getattr(var, "units", None)