def test_masking_adapter():
for col in (None, 'x'):
ds = TestDataset('', n=20)
ds_mask = MaskingAdapter(ds, '<', 10, col)
data_masked = ds_mask.read_ts()
data_masked2 = ds_mask.read()
nptest.assert_almost_equal(data_masked['x'].values,
np.concatenate([np.ones((10), dtype=bool),
np.zeros((10), dtype=bool)]))
nptest.assert_almost_equal(data_masked2['x'].values,
np.concatenate([np.ones((10), dtype=bool),
np.zeros((10), dtype=bool)]))
if col is None:
nptest.assert_almost_equal(
data_masked['y'].values, np.ones((20), dtype=bool))
nptest.assert_almost_equal(
data_masked2['y'].values, np.ones((20), dtype=bool))
def test_adapters_with_ascat():
ascat_data_folder = os.path.join(os.path.dirname(__file__), '..', 'test-data', 'sat', 'ascat', 'netcdf', '55R22')
ascat_grid_folder = os.path.join(os.path.dirname(__file__), '..', 'test-data', 'sat', 'ascat', 'netcdf', 'grid')
ascat_reader = AscatSsmCdr(ascat_data_folder, ascat_grid_folder, grid_filename='TUW_WARP5_grid_info_2_1.nc')
ascat_anom = AnomalyAdapter(ascat_reader, window_size=35, columns=['sm'])
data = ascat_anom.read_ts(12.891455, 45.923004)
assert data is not None
assert np.any(data['sm'].values != 0)
data = ascat_anom.read(12.891455, 45.923004)
assert data is not None
assert np.any(data['sm'].values != 0)
ascat_self = SelfMaskingAdapter(ascat_reader, '>', 0, 'sm')
data2 = ascat_self.read_ts(12.891455, 45.923004)
assert data2 is not None
assert np.all(data2['sm'].values > 0)
data2 = ascat_self.read(12.891455, 45.923004)
assert data2 is not None
assert np.all(data2['sm'].values > 0)
ascat_mask = MaskingAdapter(ascat_reader, '>', 0, 'sm')
data3 = ascat_mask.read_ts(12.891455, 45.923004)
assert data3 is not None
assert np.any(data3['sm'].values)
data3 = ascat_mask.read(12.891455, 45.923004)
assert data3 is not None
assert np.any(data3['sm'].values)
ascat_clim = AnomalyClimAdapter(ascat_reader, columns=['sm'])
data4 = ascat_clim.read_ts(12.891455, 45.923004)
assert data4 is not None
assert np.any(data['sm'].values != 0)
data4 = ascat_clim.read(12.891455, 45.923004)
assert data4 is not None
assert np.any(data['sm'].values != 0)
