def test_get_day_index(self):
rd = self.get_request_dataset()
dates = rd.ds.temporal.value
qdt = QEDDynamicPercentileThreshold()
di = qdt._get_day_index_(dates)
self.assertEqual((di['index'] == 366).sum(), 1)
return (di)
def test_get_geometries_with_percentiles(self):
bin_directory = '/home/local/WX/ben.koziol/links/project/ocg/bin/QED_2013_dynamic_percentiles'
qdt = QEDDynamicPercentileThreshold()
percentiles = [90, 92.5, 95, 97.5]
shp_keys = ['qed_city_centroids', 'state_boundaries', 'us_counties']
variables = ['tmin', 'tmax']
for percentile, shp_key, variable in itertools.product(
percentiles, shp_keys, variables):
ret = qdt._get_geometries_with_percentiles_(
variable, shp_key, bin_directory, percentile)
self.assertTrue(len(ret) >= 1, msg=(percentile, shp_key, len(ret)))
def test_get_dynamic_index(self):
di = self.test_get_day_index()
qdt = QEDDynamicPercentileThreshold()
dyidx = map(qdt._get_dynamic_index_, di.flat)
def test_is_leap_year(self):
years = np.array([1995, 1996, 1997])
qdt = QEDDynamicPercentileThreshold()
is_leap = map(qdt._get_is_leap_year_, years)
self.assertNumpyAll(is_leap, [False, True, False])
