test_nwm_data_time_from_dataset.py 文件源码

def test_time_from_variable():
    '''Should read date from time variable.'''

    tempdir = tempfile.gettempdir()
    nc_file = os.path.join(tempdir, 'test_time_from_variable.nc')
    date_obj = parser.parse('2017-04-29 04:00:00')
    units = 'minutes since 1970-01-01 00:00:00 UTC'
    nc_date = round(date2num(date_obj, units))
    with Dataset(nc_file, 'w') as nc:
        dim = nc.createDimension('time', 1)
        time_var = nc.createVariable('time', 'i', ('time',))
        time_var[:] = [nc_date]
        time_var.units = units
    with Dataset(nc_file, 'r') as nc:
        expected = date_obj.replace(tzinfo=pytz.utc)
        returned = nwm_data.time_from_dataset(nc)
        assert expected == returned
    os.remove(nc_file)