Function createTestTrial

master/internal/api_trials_intg_test.go:43–69

(
	t *testing.T, api *apiServer, curUser model.User,
)

var inferenceMetricGroup = "inference"

func createTestTrial(
	t *testing.T, api *apiServer, curUser model.User,
) (*model.Trial, *model.Task) {
	exp := createTestExpWithProjectID(t, api, curUser, 1)

	requestID := model.NewRequestID(rand.Reader)
	task := &model.Task{
		TaskType:   model.TaskTypeTrial,
		LogVersion: model.TaskLogVersion1,
		StartTime:  time.Now(),
		TaskID:     trialTaskID(exp.ID, requestID),
	}
	require.NoError(t, db.AddTask(context.TODO(), task))

	trial := &model.Trial{
		StartTime:    time.Now(),
		RequestID:    &requestID,
		State:        model.PausedState,
		ExperimentID: exp.ID,
	}
	require.NoError(t, db.AddTrial(context.TODO(), trial, task.TaskID))

	// Return trial exactly the way the API will generally get it.
	outTrial, err := db.TrialByID(context.TODO(), trial.ID)
	require.NoError(t, err)
	return outTrial, task
}

func createTestTrialWithMetrics(
	ctx context.Context, t *testing.T, api *apiServer, curUser model.User, includeBatchMetrics bool,