}
}
func TestLoadDataFilesCertaintyPruning(t *testing.T) {
	// Create a temp directory with test .data files
	tempDir := t.TempDir()
	// File 1: certainty 70
	file1 := `[{"Operator": "test"}, [0, 10, 100, 1000, [{"species": "Kiwi", "certainty": 70}]]]`
	if err := os.WriteFile(filepath.Join(tempDir, "file1.data"), []byte(file1), 0644); err != nil {
		t.Fatal(err)
	}
	// File 2: certainty 100
	file2 := `[{"Operator": "test"}, [0, 10, 100, 1000, [{"species": "Kiwi", "certainty": 100}]]]`
	if err := os.WriteFile(filepath.Join(tempDir, "file2.data"), []byte(file2), 0644); err != nil {
		t.Fatal(err)
	}
	// Filter by certainty 100 - should load only file2
	config := ClassifyConfig{Folder: tempDir, Certainty: 100}
	state, err := LoadDataFiles(config)
	if err != nil {
		t.Fatal(err)
	}
	if len(state.DataFiles) != 1 {
		t.Errorf("Certainty=100: expected 1 file, got %d", len(state.DataFiles))

	if state.TotalSegments() != 1 {
		t.Errorf("Certainty=100: expected 1 segment, got %d", state.TotalSegments())
	}
	// CurrentSegment should work (not nil) because file1 was pruned
	seg := state.CurrentSegment()
	if seg == nil {
		t.Error("CurrentSegment should not be nil after pruning")
	}

func TestCertaintyPruning(t *testing.T) {
	// Simulate the bug: first file has no matching certainty segments
	df1 := &utils.DataFile{
		FilePath: "/test/file1.data",
		Segments: []*utils.Segment{
			{
				StartTime: 0,
				EndTime:   10,
				Labels: []*utils.Label{
					{Species: "Kiwi", Filter: "model-1.0", Certainty: 70},
				},
			},
		},
	}
	df2 := &utils.DataFile{
		FilePath: "/test/file2.data",
		Segments: []*utils.Segment{
			{
				StartTime: 0,
				EndTime:   10,
				Labels: []*utils.Label{
					{Species: "Kiwi", Filter: "model-1.0", Certainty: 100},
				},
			},
		},
	}
	// Without pruning (old bug): file1 is first, has no certainty=100 segments
	// CurrentSegment() would return nil even though TotalSegments() > 0
	state := &ClassifyState{
		Config:    ClassifyConfig{Certainty: 100},
		DataFiles: []*utils.DataFile{df1, df2},
	}
	// TotalSegments should be 1 (only file2 has certainty 100)
	if got := state.TotalSegments(); got != 1 {
		t.Errorf("Certainty=100: expected 1 segment, got %d", got)
	}
	// CurrentSegment should work if files are properly pruned
	// Note: this test assumes LoadDataFiles does the pruning
	// Here we test the state after manual construction
}

	if config.Filter != "" || config.Species != "" {

	if config.Filter != "" || config.Species != "" || config.Certainty >= 0 {