754642db5ff35be98e4b6fc3a17823482881cb27,tests/storages_tests/test_storages.py,,test_get_all_study_summaries,#Any#,546

Before Change

def test_get_all_study_summaries(storage_init_func):
    // type: (Callable[[], BaseStorage]) -> None

    storage = storage_init_func()
    study_id = storage.create_new_study()

    storage.set_study_direction(study_id, StudyDirection.MINIMIZE)

    datetime_1 = datetime.now()

    // Set up trial 1.
    _create_new_trial_with_example_trial(
        storage, study_id, EXAMPLE_DISTRIBUTIONS, EXAMPLE_TRIALS[0]
    )

    datetime_2 = datetime.now()

    // Set up trial 2.
    trial_id_2 = storage.create_new_trial(study_id)
    storage.set_trial_value(trial_id_2, 2.0)

    for key, value in EXAMPLE_ATTRS.items():
        storage.set_study_user_attr(study_id, key, value)

    summaries = storage.get_all_study_summaries()

    assert len(summaries) == 1
    assert summaries[0]._study_id == study_id
    assert summaries[0].study_name == storage.get_study_name_from_id(study_id)
    assert summaries[0].direction == StudyDirection.MINIMIZE
    assert summaries[0].user_attrs == EXAMPLE_ATTRS
    assert summaries[0].n_trials == 2
    assert summaries[0].datetime_start is not None
    assert datetime_1 < summaries[0].datetime_start < datetime_2
    _check_example_trial_static_attributes(summaries[].best_trial, EXAMPLE_TRIALS[0])


@parametrize_storage

After Change

@pytest.mark.parametrize("storage_mode", STORAGE_MODES)
def test_get_all_study_summaries(storage_mode: str) -> None:

    with StorageSupplier(storage_mode) as storage:
        expected_summaries, _ = _setup_studies(storage, n_study=10, n_trial=10, seed=46)
        summaries = storage.get_all_study_summaries()
        assert len(summaries) == len(expected_summaries)
        for summary in summaries:
            expected_summary = expected_summaries[summary.study_id]
            assert summary.study_id == expected_summary.study_id
            assert summary.direction == expected_summary.direction
            assert summary.datetime_start == expected_summary.datetime_start
            assert summary.study_name == expected_summary.study_name
            assert summary.n_trials == expected_summary.n_trials
            assert summary.user_attrs == expected_summary.user_attrs
            assert summary.system_attrs == expected_summary.system_attrs
            if expected_summary.best_trial is not None:
                assert summary.best_trial == expected_summary.best_trial


@pytest.mark.parametrize("storage_mode", STORAGE_MODES)
def test_get_trial(storage_mode: str) -> None:

    with StorageSupplier(storage_mode) as storage:

Instances