TST Adjusts rtol for test_lda_predict
@@ -2,7 +2,6 @@ import numpy as np
import pytest
-from numpy.testing import assert_allclose
from scipy import linalg
from sklearn.exceptions import ChangedBehaviorWarning
@@ -76,8 +75,8 @@ def test_lda_predict():
assert_array_equal((y_proba_pred1[:, 1] > 0.5) + 1, y,
'solver %s' % solver)
y_log_proba_pred1 = clf.predict_log_proba(X1)
- assert_array_almost_equal(np.exp(y_log_proba_pred1), y_proba_pred1,
- 8, 'solver %s' % solver)
+ assert_allclose(np.exp(y_log_proba_pred1), y_proba_pred1,
+ rtol=1e-6, err_msg='solver %s' % solver)
# Primarily test for commit 2f34950 -- "reuse" of priors
y_pred3 = clf.fit(X, y3).predict(X)
