def test_convert_units(self, tasmax_series):
tx = np.zeros(366)
tx[:10] = np.array([29, 31, 31, 31, 29, 31, 31, 31, 31, 31])
tx = tasmax_series(tx, start='1/1/2000')
tx.attrs['units'] = 'C'
hwi = atmos.heat_wave_index(tx, freq='YS')
np.testing.assert_array_equal(hwi, [10])
def test_convert_units(self, tasmax_series):
tx = np.zeros(366)
tx[:10] = np.array([29, 31, 31, 31, 29, 31, 31, 31, 31, 31])
tx = tasmax_series(tx, start="1/1/2000")
tx.attrs["units"] = "C"
hwi = atmos.heat_wave_index(tx, freq="YS")
np.testing.assert_array_equal(hwi, [10])
def test_nan_presence(self, tasmax_series):
tx = np.zeros(366)
tx[:10] = np.array([29, 31, 31, 31, 29, 31, 31, 31, 31, 31])
tx[-1] = np.nan
tx = tasmax_series(tx + K2C, start="1/1/2000")
hwi = atmos.heat_wave_index(tx, freq="YS")
np.testing.assert_array_equal(hwi, [np.nan])
def test_simple(self, tasmax_series):
tx = np.zeros(366)
tx[:10] = np.array([29, 31, 31, 31, 29, 31, 31, 31, 31, 31])
tx = tasmax_series(tx + K2C, start='1/1/2000')
hwi = atmos.heat_wave_index(tx, freq='YS')
np.testing.assert_array_equal(hwi, [10])
