def test_gen():
 tpot_obj = TPOTClassifier()
 pipeline = tpot_obj._gen_grow_safe(tpot_obj._pset, 1, 3)
 assert (len(pipeline) > 1)
 assert (pipeline[0].ret == Output_DF)

def test_gen ( ) : tpot_obj = TPOTClassifier ( ) pipeline = tpot_obj . _gen_grow_safe ( tpot_obj . _pset , 1 , 3 ) assert ( len ( pipeline ) > 1 ) assert ( pipeline [ 0 ] . ret == Output_DF )
assert that tpots gen_grow_safe function returns a pipeline of expected structure .