fix test: functionNames contains symbols

GrammaTech · Nov 14, 2023 · 867e283 · 867e283
1 parent 46a7e49
commit 867e283
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/function_inference_test.py b/tests/function_inference_test.py
@@ -405,7 +405,10 @@ def test_code_exports_are_functions(self):
             self.assertTrue(disassemble("ex.dll", format="--ir")[0])
             ir = gtirb.IR.load_protobuf("ex.dll.gtirb")
             module = ir.modules[0]
-            functionNames = set(module.aux_data["functionNames"].data.values())
+            functionNames = {
+                sym.name
+                for sym in module.aux_data["functionNames"].data.values()
+            }
             self.assertIn("print_ok1", functionNames)
             self.assertIn("print_ok2", functionNames)
             self.assertIn("print_ok3", functionNames)