def test_mask(): df = pd.read_csv(get_path('test_data.csv'), header=[0, 1]) result = mask.mask(df, get_path('mask.csv')) assert result['wvl'].columns[0] == 586.049 assert result['wvl'].columns[-1] == 589.869 assert result['wvl'].shape == (103, 18) assert result['masked'].shape == (103, 26)
def mask(self, maskfile, maskvar): self.df = mask.mask(self.df, maskfile, maskvar=maskvar)