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