Skip to content

Commit

Permalink
add rerun to test_fit_5
Browse files Browse the repository at this point in the history
  • Loading branch information
weixuanfu committed Nov 5, 2019
1 parent 0def2d3 commit 73cec4e
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/tpot_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -970,13 +970,13 @@ def test_fit_4():


def test_fit_5():
"""Assert that the TPOT fit function provides an optimized pipeline with max_time_mins of 2 second."""
"""Assert that the TPOT fit function provides an optimized pipeline with max_time_mins of 2 second with warm_start=True."""
tpot_obj = TPOTClassifier(
random_state=42,
population_size=2,
generations=None,
verbosity=0,
max_time_mins=2/60.,
max_time_mins=3/60.,
config_dict='TPOT light',
warm_start=True
)
Expand All @@ -990,6 +990,10 @@ def test_fit_5():
assert tpot_obj._pop != []
assert isinstance(tpot_obj._optimized_pipeline, creator.Individual)
assert not (tpot_obj._start_datetime is None)
# rerun it
tpot_obj.fit(training_features, training_target)
assert tpot_obj._pop != []



def test_fit_6():
Expand Down

0 comments on commit 73cec4e

Please sign in to comment.