Python FixedIterationsStoppingCondition.should_stop Examples

Example #1

def test_fixed_iteration_stopping_condition():
    stopping_condition = FixedIterationsStoppingCondition(5)
    loop_state_mock = mock.create_autospec(LoopState)
    loop_state_mock.iteration = 0

    assert(stopping_condition.should_stop(loop_state_mock) is False)

    loop_state_mock = mock.create_autospec(LoopState)
    loop_state_mock.iteration = 5

    assert(stopping_condition.should_stop(loop_state_mock) is True)

Example #2

File: test_stopping_conditions.py Project: EmuKit/emukit

def test_fixed_iteration_stopping_condition():
    n_iterations = 5

    stopping_condition = FixedIterationsStoppingCondition(n_iterations)
    loop_state_mock = mock.create_autospec(LoopState)

    loop_state_mock.iteration = 0
    assert stopping_condition.should_stop(loop_state_mock) is False

    loop_state_mock.iteration = n_iterations - 1
    assert stopping_condition.should_stop(loop_state_mock) is False

    loop_state_mock.iteration = n_iterations
    assert stopping_condition.should_stop(loop_state_mock) is True