hy-smi_test.go

package gpu

import (
	"encoding/json"
	"os"
	"strings"
	"testing"
)

func TestGetHYVersionInfo(t *testing.T) {
	ver, err := GetHYVersionInfo()
	if err != nil {
		t.Errorf("error get hygon dcu version info: %v", err)
	}
	t.Logf("version: %+v", ver)
}

func TestGetSMIAllOutput(t *testing.T) {
	info, err := GetSMIAllOutput()
	if err != nil {
		t.Errorf("error get hygon dcu device info: %v", err)
	}
	for _, i := range info {
		t.Logf("%+v", i)
	}
}

func TestGetRunningInfo(t *testing.T) {
	info, err := GetRunningInfo()
	if err != nil {
		t.Errorf("error get hygon dcu running info: %v", err)
	}
	for _, i := range info {
		t.Logf("%+v", i)
	}
}

func TestGetDCUPidInfo(t *testing.T) {
	info, err := GetDCUPidInfo()
	if err != nil {
		t.Errorf("error get hygon dcu process: %v", err)
	}
	for _, i := range info {
		t.Logf("%+v", i)
	}
}

func TestJson(t *testing.T) {
	str := "['0','2']"
	s := make([]string, 0)
	err := json.Unmarshal([]byte(strings.ReplaceAll(str, "'", `"`)), &s)
	if err != nil {
		t.Errorf("error unmarshal: %v", err)
	}
	t.Logf("%+v", s)
}

func TestParseEcc(t *testing.T) {
	bytes, err := os.ReadFile("../test-data/hy-smi-ecc.log")
	if err != nil {
		t.Error(err)
	}
	str := strings.TrimSpace(strings.Trim(string(bytes), "\n"))
	info, err := parseEccInfo(str)
	if err != nil {
		t.Error(err)
	}
	t.Logf("%+v", info)
}

func TestGetDCUMemInfo(t *testing.T) {
	i, err := GetDCUMemInfo()
	if err != nil {
		t.Error(err)
	}
	t.Logf("%+v", i)
}