From 52b51a03248f9baceb946ccc4eaa3ea70ece22f9 Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Fri, 20 Feb 2026 13:42:33 +0800
Subject: [PATCH] test fixes from rdna4 sqtt (#14902)

---
 test/amd/test_sqttmap.py       | 2 +-
 test/testextra/test_cfg_viz.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/amd/test_sqttmap.py b/test/amd/test_sqttmap.py
index f8e388cbab..b7c760fd55 100644
--- a/test/amd/test_sqttmap.py
+++ b/test/amd/test_sqttmap.py
@@ -30,7 +30,7 @@ def rocprof_inst_traces_match(sqtt, prg, target):
     rocprof_inst = next(rwaves_iter[info.wave][0])
     ref_pc = rocprof_inst.pc-prg.base
     # always check pc matches
-    assert ref_pc == info.pc, f"pc mismatch {ref_pc}:{disasm_map[rocprof_inst.pc][0]} != {info.pc}:{disasm(info.inst)}"
+    assert ref_pc == info.pc, f"pc mismatch {ref_pc}:{disasm_map[rocprof_inst.pc]} != {info.pc}:{disasm(info.inst)}"
     # special handling for s_endpgm, it marks the wave completion.
     if info.inst == s_endpgm():
       completed_wave = list(rwaves_iter[info.wave].pop(0))
diff --git a/test/testextra/test_cfg_viz.py b/test/testextra/test_cfg_viz.py
index bf56f17a28..c6306482e6 100644
--- a/test/testextra/test_cfg_viz.py
+++ b/test/testextra/test_cfg_viz.py
@@ -30,7 +30,7 @@ class TestCfg(unittest.TestCase):
   def setUp(self):
     self.arch = Device["AMD"].arch
     if not any(self.arch.startswith(a) for a in {"gfx11", "gfx12"}):
-      self.skipTest(f"tests written for RDNA, got arch {arch}")
+      self.skipTest(f"tests written for RDNA, got arch {self.arch}")
 
   def test_simple(self):
     k = Kernel(arch=Device["AMD"].arch)