Skip to content

Commit b0f1849

Browse files
authored
fix: replace np.Inf with np.inf in tests as numpy2.0 drops np.Inf alias (#98)
1 parent 9095221 commit b0f1849

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

mne_features/tests/test_univariate.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -216,21 +216,21 @@ def test_pow_freq_bands_no_norm():
216216

217217
def test_pow_freq_bands_norm_log():
218218
_power_sum = power_sin1 + power_sin2
219-
expected = np.array([-np.Inf, 10 * np.log10(power_sin1 / _power_sum),
220-
-np.Inf, -np.Inf,
219+
expected = np.array([-np.inf, 10 * np.log10(power_sin1 / _power_sum),
220+
-np.inf, -np.inf,
221221
10 * np.log10(power_sin2 / _power_sum)])
222222
result = compute_pow_freq_bands(
223223
sfreq, data_sin, normalize=True, log=True, psd_method='fft')
224-
result[result < -250] = -np.Inf
224+
result[result < -250] = -np.inf
225225
assert_almost_equal(result, expected)
226226

227227

228228
def test_pow_freq_bands_no_norm_log():
229-
expected = np.array([-np.Inf, 10 * np.log10(power_sin1), -
230-
np.Inf, -np.Inf, 10 * np.log10(power_sin2)])
229+
expected = np.array([-np.inf, 10 * np.log10(power_sin1), -
230+
np.inf, -np.inf, 10 * np.log10(power_sin2)])
231231
result = compute_pow_freq_bands(sfreq, data_sin, normalize=False, log=True,
232232
psd_method='fft')
233-
result[result < -250] = -np.Inf
233+
result[result < -250] = -np.inf
234234
assert_almost_equal(result, expected)
235235

236236

0 commit comments

Comments
 (0)