Function TestRunBatch_ProgressCallback

pkg/evals/batch_test.go:124–176

TestRunBatch_ProgressCallback 测试进度回调

(t *testing.T)

// TestRunBatch_ProgressCallback 测试进度回调
func TestRunBatch_ProgressCallback(t *testing.T) {
	mockProvider := &MockProvider{
		response: `{"score": 0.75, "reason": "进度测试"}`,
	}

	testCases := make([]*BatchTestCase, 5)
	for i := range 5 {
		testCases[i] = &BatchTestCase{
			ID:    string(rune('A' + i)),
			Input: &TextEvalInput{Answer: "测试"},
		}
	}

	scorers := []Scorer{
		NewFaithfulnessScorer(mockProvider),
	}

	var progressCalls int
	var lastCompleted, lastTotal int

	cfg := &BatchConfig{
		TestCases:   testCases,
		Scorers:     scorers,
		Concurrency: 2,
		ProgressCallback: func(completed, total int) {
			progressCalls++
			lastCompleted = completed
			lastTotal = total
		},
	}

	result, err := RunBatch(context.Background(), cfg)
	if err != nil {
		t.Fatalf("RunBatch() error = %v", err)
	}

	// 验证进度回调被调用
	if progressCalls == 0 {
		t.Error("ProgressCallback was never called")
	}

	if lastCompleted != 5 {
		t.Errorf("Last completed = %d, want 5", lastCompleted)
	}

	if lastTotal != 5 {
		t.Errorf("Last total = %d, want 5", lastTotal)
	}

	if result.Summary.SuccessfulCases != 5 {
		t.Errorf("SuccessfulCases = %d, want 5", result.Summary.SuccessfulCases)
	}
}

// TestRunBatch_StopOnError 测试遇错停止
func TestRunBatch_StopOnError(t *testing.T) {