package blender

import "testing"

func TestParseRocmAgentArch(t *testing.T) {
	input := `ROCk module is loaded
Agent 1
  Name:                    gfx1151
  Marketing Name:          AMD Radeon Graphics
`
	arch, ok := parseRocmAgentArch(input)
	if !ok {
		t.Fatal("expected arch to be parsed")
	}
	if arch != "gfx1151" {
		t.Fatalf("arch = %q, want gfx1151", arch)
	}
}