def test_3d_data_with_nans(self, units, factor, chunks):
# test with 3d data
pr1 = open_dataset(self.nc_file).pr
pr2 = open_dataset(self.nc_file, chunks=chunks).pr
pr2.values *= factor
pr2.attrs["units"] = units
# put a nan somewhere
pr2.values[10, 1, 0] = np.nan
pr1.values[10, 1, 0] = np.nan
wind = 3
out1 = atmos.max_n_day_precipitation_amount(pr1,
window=wind,
freq="MS")
out2 = atmos.max_n_day_precipitation_amount(pr2,
window=wind,
freq="MS")
np.testing.assert_array_almost_equal(out1, out2, 3)
x1 = pr1[:31, 0, 0].values * 86400
df = pd.DataFrame({"pr": x1})
rx3 = df.rolling(wind).sum().max()
assert np.allclose(rx3, out1.values[0, 0, 0])
assert np.isnan(out1.values[0, 1, 0])
assert np.isnan(out1.values[0, -1, -1])
def test_3d_data_with_nans(self):
# test with 3d data
pr = xr.open_dataset(self.nc_file).pr
prMM = xr.open_dataset(self.nc_file).pr
prMM.values *= 86400.0
prMM.attrs["units"] = "mm/day"
# put a nan somewhere
prMM.values[10, 1, 0] = np.nan
pr.values[10, 1, 0] = np.nan
wind = 3
out1 = atmos.max_n_day_precipitation_amount(pr, window=wind, freq="MS")
out2 = atmos.max_n_day_precipitation_amount(prMM,
window=wind,
freq="MS")
# test kg m-2 s-1
pr.attrs["units"] = "kg m-2 s-1"
out3 = atmos.max_n_day_precipitation_amount(pr, window=wind, freq="MS")
np.testing.assert_array_almost_equal(out1, out2, 3)
np.testing.assert_array_almost_equal(out1, out3, 3)
x1 = prMM[:31, 0, 0].values
df = pd.DataFrame({"pr": x1})
rx3 = df.rolling(wind).sum().max()
assert np.allclose(rx3, out1.values[0, 0, 0])
assert np.isnan(out1.values[0, 1, 0])
assert np.isnan(out1.values[0, -1, -1])