Update test cases

Author: srinivasreddy

Lib/idlelib/idle_test/test_warning.py

@@ -72,34 +72,40 @@ def test_shell_show(self):
                     'Test', UserWarning, 'test_warning.py', 99, f, 'Line of code')
             self.assertEqual(shellmsg.splitlines(), f.getvalue().splitlines())
 
+
 class TestDeprecatedHelp(unittest.TestCase):
-    def setUp(self):
-        self.module = ModuleType("testmodule")
-        self.code = r"""\
+
+    def test_help_output(self):
+        # Capture the help output
+        import io
+        from contextlib import redirect_stdout
+        code = r"""\
 from warnings import deprecated
 @deprecated("Test")
 class A:
     pass
 """
-        exec(self.code, self.module.__dict__)
-        sys.modules["testmodule"] = self.module
+        subclass_code = r"""\
+from warnings import deprecated
+@deprecated("Test")
+class A:
+    pass
 
-    def tearDown(self):
-        if "testmodule" in sys.modules:
+class B(A):
+    pass
+b = B()
+"""
+        module = ModuleType("testmodule")
+        for kode in (code,subclass_code):
+            exec(kode, module.__dict__)
+            sys.modules["testmodule"] = module
+            f = io.StringIO()
+            with redirect_stdout(f):
+                help(module)
+            help_output = f.getvalue()
+            self.assertIn("Help on module testmodule:", help_output)
             del sys.modules["testmodule"]
 
-    def test_help_output(self):
-        # Capture the help output
-        import io
-        from contextlib import redirect_stdout
-
-        f = io.StringIO()
-        with redirect_stdout(f):
-            help(self.module)
-
-        help_output = f.getvalue()
-
-        self.assertIn("Help on module testmodule:", help_output)
 
 if __name__ == '__main__':
     unittest.main(verbosity=2)