Describe the feature or idea you want to propose

I'm not sure that we actually test that the max value in predict_proba matches the prediction in predict. We should do IMO. This can be added to the already present "output" tests in the general testing. I think this would only be relevant for classification and clustering.

Describe your proposed solution

Add a test that compares predict and predict_proba output in the testing module checks.

Describe alternatives you've considered, if relevant

No response

Additional context

No response