Mocks inherit from BaseEstimator