79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package render
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
)
|
|
|
|
func Test_gpuSpec(t *testing.T) {
|
|
uu := map[string]struct {
|
|
capacity v1.ResourceList
|
|
allocatable v1.ResourceList
|
|
e string
|
|
}{
|
|
"empty": {
|
|
e: NAValue,
|
|
},
|
|
|
|
"nvidia": {
|
|
capacity: v1.ResourceList{
|
|
v1.ResourceName("nvidia.com/gpu"): resource.MustParse("2"),
|
|
},
|
|
allocatable: v1.ResourceList{
|
|
v1.ResourceName("nvidia.com/gpu"): resource.MustParse("4"),
|
|
},
|
|
e: "2/4 (nvidia)",
|
|
},
|
|
|
|
"intel": {
|
|
capacity: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/i915"): resource.MustParse("2"),
|
|
},
|
|
allocatable: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/i915"): resource.MustParse("4"),
|
|
},
|
|
e: "2/4 (intel)",
|
|
},
|
|
|
|
"amd": {
|
|
capacity: v1.ResourceList{
|
|
v1.ResourceName("amd.com/gpu"): resource.MustParse("2"),
|
|
},
|
|
allocatable: v1.ResourceList{
|
|
v1.ResourceName("amd.com/gpu"): resource.MustParse("4"),
|
|
},
|
|
e: "2/4 (amd)",
|
|
},
|
|
|
|
"toast-cap": {
|
|
capacity: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/iBOZO"): resource.MustParse("2"),
|
|
},
|
|
allocatable: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/i915"): resource.MustParse("4"),
|
|
},
|
|
e: NAValue,
|
|
},
|
|
|
|
"toast-alloc": {
|
|
capacity: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/i915"): resource.MustParse("2"),
|
|
},
|
|
allocatable: v1.ResourceList{
|
|
v1.ResourceName("gpu.intel.com/iBOZO"): resource.MustParse("4"),
|
|
},
|
|
e: NAValue,
|
|
},
|
|
}
|
|
|
|
for k, u := range uu {
|
|
t.Run(k, func(t *testing.T) {
|
|
var n Node
|
|
assert.Equal(t, u.e, n.gpuSpec(u.capacity, u.allocatable))
|
|
})
|
|
}
|
|
}
|