package slicex

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestSumInt(t *testing.T) {
	testCases := []struct {
		name     string
		input    []int
		expected int
	}{
		{
			name:     "Empty ",
			input:    []int{},
			expected: 0,
		},
		{
			name:     "One Element ",
			input:    []int{1},
			expected: 1,
		},
		{
			name:     "Multiple Elements ",
			input:    []int{1, 2, 3, 4, 5},
			expected: 15,
		},
		{
			name:     "Negative Numbers",
			input:    []int{-1, -2, -3},
			expected: -6,
		},
		{
			name:     "Mix of Positive and Negative",
			input:    []int{1, -2, 3, -4, 5},
			expected: 3,
		},
		{
			name:     "Large Numbers",
			input:    []int{1000000000, 2000000000, 3000000000},
			expected: 6000000000,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			result := Sum(tc.input)
			assert.Equal(t, tc.expected, result)
		})
	}
}

func TestSumInt8(t *testing.T) {
	testCases := []struct {
		name     string
		input    []int8
		expected int8
	}{
		{
			name:     "Empty ",
			input:    []int8{},
			expected: 0,
		},
		{
			name:     "One Element ",
			input:    []int8{1},
			expected: 1,
		},
		{
			name:     "Multiple Elements ",
			input:    []int8{1, 2, 3, 4, 5},
			expected: 15,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			result := Sum(tc.input)
			assert.Equal(t, tc.expected, result)
		})
	}
}

// 测试SumInt16函数
func TestSumInt16(t *testing.T) {
	tests := []struct {
		name  string
		slice []int16
		want  int16
	}{
		{"empty slice", []int16{}, 0},
		{"one element slice", []int16{1}, 1},
		{"multiple element slice", []int16{1, 2, 3}, 6},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			res := Sum(tc.slice)
			assert.Equal(t, tc.want, res)
		})
	}
}

// 测试SumInt32函数
func TestSumInt32(t *testing.T) {
	tests := []struct {
		name  string
		slice []int32
		want  int32
	}{
		{"empty slice", []int32{}, 0},
		{"one element slice", []int32{1}, 1},
		{"multiple element slice", []int32{1, 2, 3}, 6},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			res := Sum(tt.slice)
			assert.Equal(t, tt.want, res)
		})
	}
}

// 测试SumInt64函数
func TestSumInt64(t *testing.T) {
	tests := []struct {
		name  string
		slice []int64
		want  int64
	}{
		{"empty slice", []int64{}, 0},
		{"one element slice", []int64{1}, 1},
		{"multiple element slice", []int64{1, 2, 3}, 6},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			res := Sum(tt.slice)
			assert.Equal(t, tt.want, res)
		})
	}
}

// 测试SumFloat32函数
func TestSumFloat32(t *testing.T) {
	tests := []struct {
		name  string
		slice []float32
		want  float32
	}{
		{"empty slice", []float32{}, 0},
		{"one element slice", []float32{1.0}, 1.0},
		{"multiple element slice", []float32{1.0, 2.0, 3.0}, 6.0},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			res := Sum(tt.slice)
			assert.Equal(t, tt.want, res)
		})
	}
}

// 测试SumFloat64函数
func TestSumFloat64(t *testing.T) {
	tests := []struct {
		name  string
		slice []float64
		want  float64
	}{
		{"empty slice", []float64{}, 0},
		{"one element slice", []float64{1.0}, 1.0},
		{"multiple element slice", []float64{1.0, 2.0, 3.0}, 6.0},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			res := Sum(tt.slice)
			assert.Equal(t, tt.want, res)
		})
	}
}