8ebe10a8d631b8f06b348d4dcb2f4763b824fe32,tests/test_trial.py,,test_trial_should_prune,#,207

Before Change

    study_mock = MagicMock()
    study_mock.study_id = study_id
    study_mock._storage.get_trial.return_value.\
        intermediate_values.keys.return_value = [1, 2, 3, 4, 5]
    study_mock.pruner.prune.return_value = True

    trial = Trial(study_mock, trial_id)  // type: ignore
    study_mock.reset_mock()

After Change

    // type: () -> None

    pruner = DeterministicPruner(True)
    study = create_study(pruner=pruner)
    trial = Trial(study, study._storage.create_new_trial_id(study.study_id))
    trial.report(1, 1)
    assert trial.should_prune()

Instances