From: Juhi Chandalia <jkchandalia@gmail.com>
Date: Fri, 28 Apr 2023 11:11:21 +0000 (-0600)
Subject: GH-103804: Add test for dis.disco (#103901)
X-Git-Tag: v3.12.0b1~375
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=81387fe36e76438447ff2bddc36f4f5ff2c820a2;p=thirdparty%2FPython%2Fcpython.git

GH-103804: Add test for dis.disco (#103901)
---

diff --git a/Lib/test/test_dis.py b/Lib/test/test_dis.py
index bdb541eb3b56..5262c5c257cb 100644
--- a/Lib/test/test_dis.py
+++ b/Lib/test/test_dis.py
@@ -881,6 +881,13 @@ class DisTests(DisTestBase):
         self.maxDiff = None
         got = self.get_disassembly(func, depth=0)
         self.do_disassembly_compare(got, expected, with_offsets)
+        # Add checks for dis.disco
+        if hasattr(func, '__code__'):
+            got_disco = io.StringIO()
+            with contextlib.redirect_stdout(got_disco):
+                dis.disco(func.__code__)
+            self.do_disassembly_compare(got_disco.getvalue(), expected,
+                                        with_offsets)
 
     def test_opmap(self):
         self.assertEqual(dis.opmap["NOP"], 9)