test

type CheckNodeNPUByTaskTestCase struct {
	Task    *api.TaskInfo
	Name    string
	Attr    util.SchedulerJobAttr
	Node    plugin.NPUNode
	WantErr error
}

type InitMyJobPluginTestCase struct {
	Name    string
	Attr    util.SchedulerJobAttr
	Env     plugin.ScheduleEnv
	WantErr error
}

type JudgeNodeAndTaskNPUTestCase struct {
	NodeTop []int
	Name    string
	TaskNPU int
	WantErr error
}

type ReleaseAnnotationTestCase struct {
	Task     *api.TaskInfo
	WantNode *plugin.NPUNode
	Name     string
	Node     plugin.NPUNode
	PodAnno  string
	Attr     util.SchedulerJobAttr
}

type ScoreBestNPUNodesTestCase struct {
	Task     *api.TaskInfo
	Nodes    []*api.NodeInfo
	ScoreMap map[string]float64
	WantSMap map[string]float64
	Name     string
	WantErr  error
	Attr     util.SchedulerJobAttr
}

type SetMaxNodeNPUNumTestCase struct {
	Name    string
	Num     int
	WantNum int
}

type UseAnnotationTestCase struct {
	Task     *api.TaskInfo
	WantNode *plugin.NPUNode
	Name     string
	Node     plugin.NPUNode
	PodAnno  string
	Attr     util.SchedulerJobAttr
}

type ValidNPUJobTestCase struct {
	WantErr *api.ValidateResult
	Name    string
	Attr    util.SchedulerJobAttr
}