Function TestMigStrategySingleLabels

internal/lm/mig-strategy_test.go:204–417

(t *testing.T)

}

func TestMigStrategySingleLabels(t *testing.T) {
	testCases := []struct {
		description    string
		devices        []resource.Device
		expectedError  bool
		expectedLabels Labels
		isInvalid      bool
	}{
		{
			description: "no devices returns empty labels",
		},
		{
			description: "single non-mig device returns non-mig (none) labels",
			devices: []resource.Device{
				rt.NewFullGPU(),
			},
			expectedLabels: Labels{
				"nvidia.com/gpu.compute.major":    "8",
				"nvidia.com/gpu.compute.minor":    "0",
				"nvidia.com/gpu.family":           "ampere",
				"nvidia.com/gpu.count":            "1",
				"nvidia.com/gpu.replicas":         "1",
				"nvidia.com/gpu.sharing-strategy": "none",
				"nvidia.com/gpu.memory":           "300",
				"nvidia.com/gpu.product":          "MOCKMODEL",
				"nvidia.com/mig.strategy":         "single",
			},
		},
		{
			description: "multiple non-mig device returns non-mig (none) labels",
			devices: []resource.Device{
				rt.NewFullGPU(),
				rt.NewFullGPU(),
			},
			expectedLabels: Labels{
				"nvidia.com/gpu.compute.major":    "8",
				"nvidia.com/gpu.compute.minor":    "0",
				"nvidia.com/gpu.family":           "ampere",
				"nvidia.com/gpu.count":            "2",
				"nvidia.com/gpu.replicas":         "1",
				"nvidia.com/gpu.sharing-strategy": "none",
				"nvidia.com/gpu.memory":           "300",
				"nvidia.com/gpu.product":          "MOCKMODEL",
				"nvidia.com/mig.strategy":         "single",
			},
		},
		{
			description: "single mig-enabled device returns mig labels",
			devices: []resource.Device{
				rt.NewMigEnabledDevice(
					rt.NewMigDevice(1, 2, 100),
				),
			},
			expectedLabels: Labels{
				"nvidia.com/gpu.count":            "1",
				"nvidia.com/gpu.replicas":         "1",
				"nvidia.com/gpu.sharing-strategy": "none",
				"nvidia.com/gpu.memory":           "100",