Exemplo n.º 1
0
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)