Function TestStorageRange

utils/spec_store_test.go:112–135

(t *testing.T)

}

func TestStorageRange(t *testing.T) {
	s := testSpecStore()

	tests := []struct {
		name        string
		gpuType     string
		numGPUs     int
		expectedMin int
		expectedMax int
	}{
		{"a6000", "a6000", 1, 100, 500},
		{"a100xl x1", "a100xl", 1, 100, 500},
		{"a100xl x2", "a100xl", 2, 100, 1000},
		{"unknown falls back to 100-1000", "unknown", 1, 100, 1000},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			minGB, maxGB := s.StorageRange(tt.gpuType, tt.numGPUs)
			assert.Equal(t, tt.expectedMin, minGB)
			assert.Equal(t, tt.expectedMax, maxGB)
		})
	}
}

func TestNormalizeGPUType(t *testing.T) {
	s := testSpecStore()