Updated tests
Removed check for partial_fit attribute.