219e29442e44b5d16488d946574cf6030eaa4d52,tests/store/tracking/test_abstract_store.py,,test_list_run_infos,#,110

Before Change

    with mock.patch.object(AbstractStoreTestImpl, "search_runs", return_value=runs):
        store = AbstractStoreTestImpl()
        assert store.list_run_infos(experiment_id, view_type) == run_infos
        store.search_runs.assert_called_once_with([experiment_id], None, view_type)


def test_search_runs():

After Change

    run_infos = [mock.Mock()]
    runs = [mock.Mock(info=info) for info in run_infos]

    with mock.patch.object(AbstractStoreTestImpl, "search_runs",
                           return_value=PagedList(runs, None)):
        store = AbstractStoreTestImpl()
        result = store.list_run_infos(experiment_id, view_type, page_token=token)
        for i in range(len(result)):
            assert result[i] == run_infos[i]
        assert result.token is None
        store.search_runs.assert_called_once_with([experiment_id], None, view_type,
                                                  SEARCH_MAX_RESULTS_DEFAULT, None, token)


def test_search_runs():
    experiment_id = mock.Mock()
    view_type = mock.Mock()
    runs = [mock.Mock(), mock.Mock()]

Instances