def test_mask_fill(self):
arg = np.asarray([1, 1, 1])
mask = [True, False, True]
expected = np.asarray([1, 3.0, 1])
rv = mask_fill(arg, mask=mask, value=3.0)
assert (rv == expected).all()
arg = np.array([1, 1, 1])
mask = [True, False, True]
expected = np.array([1, 3.0, 1])
rv = mask_fill(arg, mask=mask, value=3.0)
assert (rv == expected).all()
def test_mask_fill_no_mask(self):
arg = np.asarray([1, 1, 1])
expected = np.asarray([1, 1, 1])
rv = mask_fill(arg)
assert (rv == expected).all()
def test_mask_fill(self):
arg = [1, 1, 1]
mask = [True, False, True]
expected = [1, 3.0, 1]
rv = mask_fill(arg, mask=mask, value=3.0)
assert type(rv) == list
assert rv == expected
arg = np.array([1, 1, 1])
mask = [True, False, True]
expected = np.array([1, 3.0, 1])
rv = mask_fill(arg, mask=mask, value=3.0)
assert type(rv) == np.ndarray
assert (rv == expected).all()
def test_mask_fill_no_mask(self):
arg = [1, 1, 1]
expected = [1, 1, 1]
rv = mask_fill(arg)
assert type(rv) == list
assert rv == expected