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)
