def test_set_params(): """Assert that set_params returns a reference to the TPOT instance""" tpot_obj = TPOTClassifier() assert tpot_obj.set_params() is tpot_obj
def test_set_params_2(): """Assert that set_params updates TPOT's instance variables""" tpot_obj = TPOTClassifier(generations=2) tpot_obj.set_params(generations=3) assert tpot_obj.generations == 3