Just use `assert_array_equal(clf.decision_function(X_test).argmax(axis=1), clf.predict(X_test))`