def test_get_high_res_images(self): provider = AirTemperatureProvider(CubeConfig(grid_width=4320, grid_height=2160, spatial_res=1 / 12, end_time=datetime(2001, 6, 1, 0, 0)), dir=SOURCE_DIR) provider.prepare() images = provider.compute_variable_images(datetime(2001, 1, 1), datetime(2001, 1, 9)) self.assertIsNotNone(images) self.assertTrue('air_temperature_2m' in images) image = images['air_temperature_2m'] self.assertEqual((2160, 4320), image.shape)
def test_get_images(self): provider = AirTemperatureProvider(CubeConfig(end_time=datetime(2001, 6, 1, 0, 0)), dir=SOURCE_DIR, resampling_order="space_first") provider.prepare() images = provider.compute_variable_images(datetime(2001, 1, 1), datetime(2001, 1, 9)) self.assertIsNotNone(images) self.assertTrue('air_temperature_2m' in images) image = images['air_temperature_2m'] self.assertEqual((720, 1440), image.shape)
def test_get_images(self): provider = AirTemperatureProvider( CubeConfig(end_time=datetime(2001, 6, 1, 0, 0)), dir=SOURCE_DIR, resampling_order="space_first") provider.prepare() images = provider.compute_variable_images(datetime(2001, 1, 1), datetime(2001, 1, 9)) self.assertIsNotNone(images) self.assertTrue('air_temperature_2m' in images) image = images['air_temperature_2m'] self.assertEqual((720, 1440), image.shape)
def test_get_high_res_images(self): provider = AirTemperatureProvider(CubeConfig(grid_width=4320, grid_height=2160, spatial_res=1 / 12, end_time=datetime( 2001, 6, 1, 0, 0)), dir=SOURCE_DIR) provider.prepare() images = provider.compute_variable_images(datetime(2001, 1, 1), datetime(2001, 1, 9)) self.assertIsNotNone(images) self.assertTrue('air_temperature_2m' in images) image = images['air_temperature_2m'] self.assertEqual((2160, 4320), image.shape)