Fixup tests

python · markshannon · Mar 7, 2025 · Mar 4, 2025 · Mar 4, 2025 · Mar 4, 2025
commit 28991037ce2b23581dab8b81a37011a133bb8f67
diff --git a/Lib/test/test_code.py b/Lib/test/test_code.py
@@ -943,7 +943,7 @@ async def afunc():
 
         self.assertEqual(
             get_line_branches(afunc),
-            [(1,2,3)])
+            [(1,1,3)])
 
 if check_impl_detail(cpython=True) and ctypes is not None:
     py = ctypes.pythonapi

diff --git a/Lib/test/test_dis.py b/Lib/test/test_dis.py
@@ -1256,6 +1256,17 @@ def test__try_compile_no_context_exc_on_error(self):
         except Exception as e:
             self.assertIsNone(e.__context__)
 
+    def test_async_for(self):
+
+        async def afunc():
+            async for letter in async_iter1:
+                l2
+            l3
+
+        expected = ""
+        self.do_disassembly_test(afunc, expected)
+
+
     @staticmethod
     def code_quicken(f):
         _testinternalcapi = import_helper.import_module("_testinternalcapi")