def test_mask_indices(fake_nc_file): nc = netCDF4.Dataset(fake_nc_file) test_var = nc.variables['time'] obj = CloudnetArray(test_var, 'test_name') obj.mask_indices([0, 1]) result = ma.array([0, 1, 2, 3, 4], mask=[1, 1, 0, 0, 0]) assert_array_equal(obj.data.data, result.data) assert_array_equal(obj.data.mask, result.mask)
def test_mask_indices(self): obj = CloudnetArray(self.time, "test_name") obj.mask_indices([0, 1]) result = ma.array([0, 1, 2, 3, 4], mask=[1, 1, 0, 0, 0]) assert_array_equal(obj.data.data, result.data) assert_array_equal(obj.data.mask, result.mask)