コード例 #1
0
 def test_get_high_res_images(self):
     provider = PrecipProvider(CubeConfig(grid_width=4320, grid_height=2160, spatial_res=1 / 12), dir=SOURCE_DIR)
     provider.prepare()
     images = provider.compute_variable_images(datetime(1996, 1, 1), datetime(1996, 1, 9))
     self.assertIsNotNone(images)
     self.assertTrue('precipitation' in images)
     image = images['precipitation']
     self.assertEqual((2160, 4320), image.shape)
コード例 #2
0
 def test_get_images(self):
     provider = PrecipProvider(CubeConfig(), dir=SOURCE_DIR)
     provider.prepare()
     images = provider.compute_variable_images(datetime(1996, 1, 1), datetime(1996, 1, 9))
     self.assertIsNotNone(images)
     self.assertTrue('precipitation' in images)
     image = images['precipitation']
     self.assertEqual((720, 1440), image.shape)
コード例 #3
0
 def test_source_time_ranges(self):
     provider = PrecipProvider(CubeConfig(), dir=SOURCE_DIR)
     provider.prepare()
     source_time_ranges = provider.source_time_ranges
     self.assertEqual(13149, len(source_time_ranges))
     self.assert_source_time_ranges(source_time_ranges[0],
                                    datetime(1979, 1, 1, 0, 0, 0, 33),
                                    datetime(1979, 1, 2, 0, 0, 0, 33),
                                    self.get_source_dir_list(SOURCE_DIR) + ['Precip.V1.720.360.1979.nc.gz'],
                                    0)
     self.assert_source_time_ranges(source_time_ranges[6],
                                    datetime(1979, 1, 7, 0, 0, 0, 33),
                                    datetime(1979, 1, 8, 0, 0, 0, 33),
                                    self.get_source_dir_list(SOURCE_DIR) + ['Precip.V1.720.360.1979.nc.gz'],
                                    6)
     self.assert_source_time_ranges(source_time_ranges[13148],
                                    datetime(2014, 12, 31, 0, 0, 0, 33),
                                    datetime(2015, 1, 1, 0, 0, 0, 33),
                                    self.get_source_dir_list(SOURCE_DIR) + ['Precip.RT.720.360.2014.nc.gz'],
                                    364)
コード例 #4
0
 def test_temporal_coverage(self):
     provider = PrecipProvider(CubeConfig(), dir=SOURCE_DIR)
     provider.prepare()
     self.assertEqual((datetime(1979, 1, 1, 0, 0, 0, 33), datetime(2015, 1, 1, 0, 0, 0, 33)),
                      provider.temporal_coverage)