def test_wet_spells_on_list(self):
"Test wet_spells"
data = self.data
durations = [4, 1, 3, 1, 1]
intensities = [10, 1, 9, 7, 5]
_test = wet_spells(data, threshold=0.254)
assert_equal(_test['durations'], durations)
assert_equal(_test['intensities'], intensities)
def test_wet_and_dry_spells(self):
"Test wet and dry spells"
data = self.data + [-3]
rainfall = hydro.time_series(data, mask=24 * [0] + [1], start_date=hydro.Date("M", "2001-01"))
threshold = 0.254
dspells = dry_spells(rainfall, threshold=threshold)
wspells = wet_spells(rainfall, threshold=threshold)
assert_equal(dspells.sum() + wspells.durations.sum(), rainfall.count())
def test_wet_spells_on_list(self):
"Test wet_spells"
data = self.data
durations = [4, 1, 3, 1, 1]
intensities = [10, 1, 9, 7, 5]
_test = wet_spells(data, threshold=0.254)
assert_equal(_test["durations"], durations)
assert_equal(_test["intensities"], intensities)
def test_wet_and_dry_spells(self):
"Test wet and dry spells"
data = self.data + [-3]
rainfall = hydro.time_series(data,
mask=24 * [0] + [1],
start_date=hydro.Date('M', '2001-01'))
threshold = 0.254
dspells = dry_spells(rainfall, threshold=threshold)
wspells = wet_spells(rainfall, threshold=threshold)
assert_equal(dspells.sum() + wspells.durations.sum(), rainfall.count())
def test_wet_spells(self):
"Test wet_spells"
rainfall = self.rainfall
dates = rainfall.dates
control_dates_onstart = dates[[0, 5, 10, 21, 23]]
control_dates_onend = dates[[3, 5, 12, 21, 23]]
durations = [4, 1, 3, 1, 1]
intensities = [10, 1, 9, 7, 5]
#
_test = wet_spells(rainfall, threshold=0.254)
assert_equal(_test['durations'], durations)
assert_equal(_test['intensities'], intensities)
assert_equal(_test.dates, control_dates_onstart)
_test = wet_spells(rainfall, threshold=0.254, start=False)
assert_equal(_test['durations'], durations)
assert_equal(_test['intensities'], intensities)
assert_equal(_test.dates, control_dates_onend)
#
_test = wet_spells(rainfall, threshold=0.254, mode='durations')
assert_equal(_test, durations)
_test = wet_spells(rainfall, threshold=0.254, mode='d')
assert_equal(_test, durations)
_test = wet_spells(rainfall, threshold=0.254, mode='intensities')
assert_equal(_test, intensities)
_test = wet_spells(rainfall, threshold=0.254, mode='i')
assert_equal(_test, intensities)
def test_wet_spells(self):
"Test wet_spells"
rainfall = self.rainfall
dates = rainfall.dates
control_dates_onstart = dates[[0, 5, 10, 21, 23]]
control_dates_onend = dates[[3, 5, 12, 21, 23]]
durations = [4, 1, 3, 1, 1]
intensities = [10, 1, 9, 7, 5]
#
_test = wet_spells(rainfall, threshold=0.254)
assert_equal(_test["durations"], durations)
assert_equal(_test["intensities"], intensities)
assert_equal(_test.dates, control_dates_onstart)
_test = wet_spells(rainfall, threshold=0.254, start=False)
assert_equal(_test["durations"], durations)
assert_equal(_test["intensities"], intensities)
assert_equal(_test.dates, control_dates_onend)
#
_test = wet_spells(rainfall, threshold=0.254, mode="durations")
assert_equal(_test, durations)
_test = wet_spells(rainfall, threshold=0.254, mode="d")
assert_equal(_test, durations)
_test = wet_spells(rainfall, threshold=0.254, mode="intensities")
assert_equal(_test, intensities)
_test = wet_spells(rainfall, threshold=0.254, mode="i")
assert_equal(_test, intensities)
