def test_checkDayOffsetLists(self): self.assertEqual(monthlist((2014, 1, 31), 1), [0]) self.assertEqual(monthlist((2014, 1, 31), 2), [0, 28]) self.assertEqual(monthlist((2014, 1, 31), 24), [ 0, 28, 59, 89, 120, 150, 181, 212, 242, 273, 303, 334, 365, 393, 424, 454, 485, 515, 546, 577, 607, 638, 668, 699 ]) return;
def test_checkDayOffsetLists(self): self.assertEqual(monthlist((2014, 1, 31), 1), [0]) self.assertEqual(monthlist((2014, 1, 31), 2), [0, 28]) self.assertEqual(monthlist((2014, 1, 31), 24), [ 0, 28, 59, 89, 120, 150, 181, 212, 242, 273, 303, 334, 365, 393, 424, 454, 485, 515, 546, 577, 607, 638, 668, 699 ]) return
def generate2D(filename, start, months, lats, lons): cdf = Dataset(filename, 'w', format='NETCDF4') cdf.createDimension('time', months) times = cdf.createVariable('time', 'f8', ('time', )) times.units = 'days since %04d-%02d-%02d 00:00:00.0' % start times.calendar = 'gregorian' times[:] = monthlist(start, months) cdf.createDimension('lat', lats) latitudes = cdf.createVariable('lat', 'f4', ('lat', )) latitudes.units = 'degrees north' latitudes[:] = latList(lats) cdf.createDimension('lon', lons) longitudes = cdf.createVariable('lon', 'f4', ('lon', )) longitudes.units = 'degrees east' longitudes[:] = lonList(lons) randoms = cdf.createVariable('random', 'i4', ('time', 'lat', 'lon', )) randoms.units = 'random values' randoms[:] = numpy.random.randint(-5, 6, (months, lats, lons, )) cdf.close() return
def generate1D(filename, start, months): cdf = Dataset(filename, 'w', format='NETCDF4') time_dim = cdf.createDimension('time', months) time_var = cdf.createVariable('time', 'f8', ('time', )) random_var = cdf.createVariable('random', 'i4', ('time', )) time_var.units = ' '.join( ['days since ', '%04d-%02d-%02d' % start, ' 00:00:00.0']) time_var.calendar = 'gregorian' random_var.units = 'random values' time_var[:] = monthlist(start, months) random_var[:] = numpy.random.randint(-5, 6, months) cdf.close() return
def generate1D(filename, start, months): cdf = Dataset(filename, 'w', format='NETCDF4') time_dim = cdf.createDimension('time', months) time_var = cdf.createVariable('time', 'f8', ('time', )) random_var = cdf.createVariable('random', 'i4', ('time', )) time_var.units = ' '.join(['days since ', '%04d-%02d-%02d' % start, ' 00:00:00.0']) time_var.calendar = 'gregorian' random_var.units = 'random values' time_var[:] = monthlist(start, months) random_var[:] = numpy.random.randint(-5,6,months) cdf.close() return