class TestDatasetFunctions(unittest.TestCase): def setUp(self): self.lat = np.array([10, 12, 14, 16, 18]) self.lon = np.array([100, 102, 104, 106, 108]) self.time = np.array([dt.datetime(2000, x, 1) for x in range(1, 13)]) flat_array = np.array(range(300)) self.value = flat_array.reshape(12, 5, 5) self.variable = 'prec' self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) def test_spatial_boundaries(self): self.assertEqual( self.test_dataset.spatial_boundaries(), (min(self.lat), max(self.lat), min(self.lon), max(self.lon))) def test_time_range(self): self.assertEqual( self.test_dataset.time_range(), (dt.datetime(2000, 1, 1), dt.datetime(2000, 12, 1))) def test_spatial_resolution(self): self.assertEqual(self.test_dataset.spatial_resolution(), (2, 2)) def test_temporal_resolution(self): self.assertEqual(self.test_dataset.temporal_resolution(), 'monthly')
class TestDatasetFunctions(unittest.TestCase): def setUp(self): self.lat = np.array([10, 12, 14, 16, 18]) self.lon = np.array([100, 102, 104, 106, 108]) self.time = np.array([dt.datetime(2000, x, 1) for x in range(1, 13)]) flat_array = np.array(range(300)) self.value = flat_array.reshape(12, 5, 5) self.variable = 'prec' self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) def test_spatial_boundaries(self): self.assertEqual( self.test_dataset.spatial_boundaries(), (min(self.lat), max(self.lat), min(self.lon), max(self.lon))) def test_time_range(self): self.assertEqual( self.test_dataset.time_range(), (dt.datetime(2000, 1, 1), dt.datetime(2000, 12, 1))) def test_spatial_resolution(self): self.assertEqual(self.test_dataset.spatial_resolution(), (2, 2)) def test_spatial_resolution_2_dim_lat_lon(self): self.lat = np.array([10, 12, 14, 16, 18, 20]) self.lon = np.array([100, 102, 104, 106, 108, 110]) self.lat = self.lat.reshape(3, 2) self.lon = self.lon.reshape(3, 2) flat_array = np.array(range(72)) self.value = flat_array.reshape(12, 3, 2) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.spatial_resolution(), (6, 6)) def test_temporal_resolution_hourly(self): self.time = np.array([dt.datetime(2000, 1, 1), dt.datetime(2000, 1, 1)]) flat_array = np.array(range(50)) self.value = flat_array.reshape(2, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'minutely') def test_temporal_resolution_monthly(self): self.assertEqual(self.test_dataset.temporal_resolution(), 'monthly') def test_temporal_resolution_daily(self): self.time = np.array([dt.datetime(2000, 3, x) for x in range(1, 31)]) flat_array = np.array(range(750)) self.value = flat_array.reshape(30, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'daily') def test_temporal_resolution_yearly(self): self.time = np.array([dt.datetime(x, 6, 1) for x in range(2000, 2015)]) flat_array = np.array(range(375)) self.value = flat_array.reshape(15, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'yearly') def test_str_(self): dataset = self.test_dataset lat_min, lat_max, lon_min, lon_max = dataset.spatial_boundaries() start, end = dataset.time_range() lat_range = "({}, {})".format(lat_min, lon_min) lon_range = "({}, {})".format(lon_min, lon_min) time_range = "({}, {})".format(start, end) formatted_repr = ( "<Dataset - name: {}, " "lat-range: {}, " "lon-range: {}, " "time_range: {}, " "var: {}, " "units: {}>" ) output = formatted_repr.format( dataset.name if dataset.name != "" else None, lat_range, lon_range, time_range, dataset.variable, dataset.units ) self.assertEqual(str(self.test_dataset), output)
class TestDatasetFunctions(unittest.TestCase): def setUp(self): self.lat = np.array([10, 12, 14, 16, 18]) self.lon = np.array([100, 102, 104, 106, 108]) self.time = np.array([dt.datetime(2000, x, 1) for x in range(1, 13)]) flat_array = np.array(range(300)) self.value = flat_array.reshape(12, 5, 5) self.variable = 'prec' self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) def test_spatial_boundaries(self): self.assertEqual( self.test_dataset.spatial_boundaries(), (min(self.lat), max(self.lat), min(self.lon), max(self.lon))) def test_time_range(self): self.assertEqual(self.test_dataset.time_range(), (dt.datetime(2000, 1, 1), dt.datetime(2000, 12, 1))) def test_spatial_resolution(self): self.assertEqual(self.test_dataset.spatial_resolution(), (2, 2)) def test_spatial_resolution_2_dim_lat_lon(self): self.lat = np.array([10, 12, 14, 16, 18, 20]) self.lon = np.array([100, 102, 104, 106, 108, 110]) self.lat = self.lat.reshape(3, 2) self.lon = self.lon.reshape(3, 2) flat_array = np.array(range(72)) self.value = flat_array.reshape(12, 3, 2) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.spatial_resolution(), (6, 6)) def test_temporal_resolution_hourly(self): self.time = np.array( [dt.datetime(2000, 1, 1), dt.datetime(2000, 1, 1)]) flat_array = np.array(range(50)) self.value = flat_array.reshape(2, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'minutely') def test_temporal_resolution_monthly(self): self.assertEqual(self.test_dataset.temporal_resolution(), 'monthly') def test_temporal_resolution_daily(self): self.time = np.array([dt.datetime(2000, 3, x) for x in range(1, 31)]) flat_array = np.array(range(750)) self.value = flat_array.reshape(30, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'daily') def test_temporal_resolution_yearly(self): self.time = np.array([dt.datetime(x, 6, 1) for x in range(2000, 2015)]) flat_array = np.array(range(375)) self.value = flat_array.reshape(15, 5, 5) self.test_dataset = Dataset(self.lat, self.lon, self.time, self.value, self.variable) self.assertEqual(self.test_dataset.temporal_resolution(), 'yearly') def test_str_(self): dataset = self.test_dataset lat_min, lat_max, lon_min, lon_max = dataset.spatial_boundaries() start, end = dataset.time_range() lat_range = "({}, {})".format(lat_min, lon_min) lon_range = "({}, {})".format(lon_min, lon_min) time_range = "({}, {})".format(start, end) formatted_repr = ("<Dataset - name: {}, " "lat-range: {}, " "lon-range: {}, " "time_range: {}, " "var: {}, " "units: {}>") output = formatted_repr.format( dataset.name if dataset.name != "" else None, lat_range, lon_range, time_range, dataset.variable, dataset.units) self.assertEqual(str(self.test_dataset), output)