def test_set_params(estimator_instance): """Check that set_params works correctly.""" estimator = estimator_instance params = estimator.get_params() assert estimator.set_params(**params) is estimator _check_set_params(estimator.__class__.__name__, estimator)
def check_set_params(Estimator): # Check set_params works correctly estimator = _construct_instance(Estimator) params = estimator.get_params() assert estimator.set_params(**params) is estimator _check_set_params(estimator.__class__.__name__, estimator)