gh-116789: Add more tests for inspect.getmembers

sobolevn · sobolevn · commit becc50f68eb6 · 2024-03-14T15:18:56.000+03:00
diff --git a/Lib/test/test_inspect/test_inspect.py b/Lib/test/test_inspect/test_inspect.py
@@ -1512,6 +1512,71 @@ def f(self):
         self.assertIn(('f', b.f), inspect.getmembers(b))
         self.assertIn(('f', b.f), inspect.getmembers(b, inspect.ismethod))
 
+    def test_getmembers_function(self):
+        def func(arg: int = 1, *, kw = 2) -> None:
+            """Docstring."""
+
+        res = dict(inspect.getmembers(func))
+        expected = {k: getattr(func, k) for k in dir(func)}
+        self.assertEqual(res, expected)
+
+    def test_getmembers_function_predicate(self):
+        def func(*, a: int = 1): ...
+
+        res = dict(
+            inspect.getmembers(
+                func, lambda value: isinstance(value, dict),
+            ),
+        )
+        expected = {
+            k: v
+            for k in dir(func)
+            if isinstance((v := getattr(func, k)), dict)
+        }
+        self.assertEqual(res, expected)
+
+    def test_getmembers_traceback(self):
+        try:
+            raise ValueError(3)
+        except ValueError as ex:
+            trace = ex.__traceback__
+
+        res = dict(inspect.getmembers(trace))
+        expected = {k: getattr(trace, k) for k in dir(trace)}
+        self.assertEqual(res, expected)
+
+    def test_getmembers_frame(self):
+        ns = {}
+        def func(a=1):
+            ns['fr'] = inspect.currentframe()
+        func()
+
+        res = dict(inspect.getmembers(ns['fr']))
+        expected = {k: getattr(ns['fr'], k) for k in dir(ns['fr'])}
+        self.assertEqual(res, expected)
+
+    def test_getmembers_code(self):
+        def func(): ...
+
+        res = dict(inspect.getmembers(func.__code__))
+        expected = {k: getattr(func.__code__, k) for k in dir(func.__code__)}
+        self.assertEqual(res, expected)
+
+    def test_getmembers_code(self):
+        async def func(): ...
+
+        coro = func()
+        self.addCleanup(coro.close)
+
+        res = dict(inspect.getmembers(coro))
+        expected = {k: getattr(coro, k) for k in dir(coro)}
+        self.assertEqual(res, expected)
+
+    def test_getmembers_builtin(self):
+        res = dict(inspect.getmembers(list))
+        expected = {k: getattr(list, k) for k in dir(list)}
+        self.assertEqual(res, expected)
+
     def test_getmembers_VirtualAttribute(self):
         class M(type):
             def __getattr__(cls, name):
