def test_freq_mask():
"""Test safe frequency masking."""
N = 10
x = np.arange(N).astype(float)
assert _freq_mask(x, 1000., fmin=0, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=0, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=None).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=-np.inf, fmax=None).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=np.inf).sum() == N
# non-uniformly spaced inputs
x = np.array([4, 10])
assert _freq_mask(x[:1], 1, fmin=10, raise_error=False).sum() == 0
assert _freq_mask(x[:1], 1, fmin=11, fmax=12, raise_error=False).sum() == 0
assert _freq_mask(x, sfreq=1, fmin=10).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=6).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=5).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=4.5001).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=4.4999).sum() == 2
assert _freq_mask(x, sfreq=1, fmin=4).sum() == 2
# degenerate cases
with pytest.raises(ValueError, match='sfreq can not be None'):
_freq_mask(x[:1], sfreq=None, fmin=3, fmax=5)
with pytest.raises(ValueError, match='No frequencies remain'):
_freq_mask(x[:1], sfreq=1, fmin=11, fmax=12)
with pytest.raises(ValueError, match='must be less than or equal to fmax'):
_freq_mask(x[:1], sfreq=1, fmin=10)
def test_freq_mask():
"""Test safe frequency masking."""
N = 10
x = np.arange(N).astype(float)
assert _freq_mask(x, 1000., fmin=0, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=0, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=N - 1).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=None).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=-np.inf, fmax=None).sum() == N
assert _freq_mask(x - 1e-10, 1000., fmin=None, fmax=np.inf).sum() == N
# non-uniformly spaced inputs
x = np.array([4, 10])
assert _freq_mask(x[:1], 1, fmin=10, raise_error=False).sum() == 0
assert _freq_mask(x[:1], 1, fmin=11, fmax=12,
raise_error=False).sum() == 0
assert _freq_mask(x, sfreq=1, fmin=10).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=6).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=5).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=4.5001).sum() == 1
assert _freq_mask(x, sfreq=1, fmin=4.4999).sum() == 2
assert _freq_mask(x, sfreq=1, fmin=4).sum() == 2
# degenerate cases
with pytest.raises(ValueError, match='sfreq can not be None'):
_freq_mask(x[:1], sfreq=None, fmin=3, fmax=5)
with pytest.raises(ValueError, match='No frequencies remain'):
_freq_mask(x[:1], sfreq=1, fmin=11, fmax=12)
with pytest.raises(ValueError, match='must be less than or equal to fmax'):
_freq_mask(x[:1], sfreq=1, fmin=10)