fix test

Author: AndPuQing

Lib/test/test_dis.py

@@ -437,33 +437,11 @@ def foo(a: int, b: str) -> str:
 __dataclass_fields__: ClassVar
 """
 
+issue_135700_class = """\
 class _B:
     2
     c: int
-
-dis_issue_135700_class = """\
-Disassembly of __annotate_func__:
-  --           COPY_FREE_VARS           1
-
- %3d           RESUME                   0
-               LOAD_FAST_BORROW         0 (format)
-               LOAD_SMALL_INT           2
-               COMPARE_OP             132 (>)
-               POP_JUMP_IF_FALSE        3 (to L1)
-               NOT_TAKEN
-               LOAD_COMMON_CONSTANT     1 (NotImplementedError)
-               RAISE_VARARGS            1
-       L1:     BUILD_MAP                0
-
- %3d           LOAD_DEREF               1 (__classdict__)
-               LOAD_FROM_DICT_OR_GLOBALS 0 (int)
-               COPY                     2
-               LOAD_CONST               1 ('c')
-
- %3d           STORE_SUBSCR
-               RETURN_VALUE
-
-""" % (_B.__firstlineno__, _B.__firstlineno__ + 2, _B.__firstlineno__)
+"""
 
 compound_stmt_str = """\
 x = 0
@@ -1160,21 +1138,25 @@ def test_bug_46724(self):
 
     def test_annotate_no_spurious_first_node_positions(self):
         # Test that __annotate__ code doesn't inherit first AST node positions
-        annotate_code = compile(issue_135700, "<string>", "exec").co_consts[1]
-        instructions = list(dis.Bytecode(annotate_code))
-
-        spurious_positions = 0
-        for instr in instructions:
-            if (instr.positions and
-                instr.positions.lineno == 1 and
-                instr.positions.col_offset == 0 and
-                instr.positions.end_col_offset == 1):
-                spurious_positions += 1
-
-        self.assertEqual(spurious_positions, 0,
-                        "No instructions should have first statement's position")
-        got = self.get_disassembly(_B, depth=1)
-        self.do_disassembly_compare(got, dis_issue_135700_class)
+        test_cases = [
+            ("module", compile(issue_135700, "<string>", "exec").co_consts[1]),
+            ("class", compile(ast.parse(issue_135700_class), "?", "exec").co_consts[0].co_consts[1])
+        ]
+
+        for case_name, annotate_code in test_cases:
+            with self.subTest(case=case_name):
+                instructions = list(dis.Bytecode(annotate_code))
+
+                spurious_positions = sum(
+                    1 for instr in instructions
+                    if (instr.positions and
+                        instr.positions.lineno == 1 and
+                        instr.positions.col_offset == 0 and
+                        instr.positions.end_col_offset == 1)
+                )
+
+                self.assertEqual(spurious_positions, 0,
+                            f"No instructions in {case_name} __annotate__ should have first statement's position")
 
     def test_kw_names(self):
         # Test that value is displayed for keyword argument names: