volcano.sh/volcano@v1.9.0/pkg/scheduler/api/helpers/helpers_test.go (about)

     1  package helpers
     2  
     3  import (
     4  	"reflect"
     5  	"testing"
     6  
     7  	v1 "k8s.io/api/core/v1"
     8  
     9  	"volcano.sh/volcano/pkg/scheduler/api"
    10  )
    11  
    12  func TestMax(t *testing.T) {
    13  	l := &api.Resource{
    14  		MilliCPU: 1,
    15  		Memory:   1024,
    16  		ScalarResources: map[v1.ResourceName]float64{
    17  			"gpu":    1,
    18  			"common": 4,
    19  		},
    20  	}
    21  	r := &api.Resource{
    22  		MilliCPU: 2,
    23  		Memory:   1024,
    24  		ScalarResources: map[v1.ResourceName]float64{
    25  			"npu":    2,
    26  			"common": 5,
    27  		},
    28  	}
    29  	expected := &api.Resource{
    30  		MilliCPU: 2,
    31  		Memory:   1024,
    32  		ScalarResources: map[v1.ResourceName]float64{
    33  			"gpu":    1,
    34  			"npu":    2,
    35  			"common": 5,
    36  		},
    37  	}
    38  	re := Max(l, r)
    39  	if !reflect.DeepEqual(expected, re) {
    40  		t.Errorf("expected: %#v, got: %#v", expected, re)
    41  	}
    42  }