FIX multiple issues in the newly added partial_fit tests.