bpo-44808: Fix test_inspect in refleak mode (GH-27544)

python · Aug 2, 2021 · 626d397 · 626d397
1 parent 414dcb1
commit 626d397
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py
@@ -587,11 +587,12 @@ def test_getsource_on_code_object(self):
 
 class TestGetsourceInteractive(unittest.TestCase):
     def tearDown(self):
-        mod.ParrotDroppings.__module__ = mod
+        mod.ParrotDroppings.__module__ = self.mod
         sys.modules['__main__'] = self.main
 
     def test_getclasses_interactive(self):
         self.main = sys.modules['__main__']
+        self.mod = mod.ParrotDroppings.__module__
         class MockModule:
             __file__ = None
         sys.modules['__main__'] = MockModule