swap testing with xml to custom class

Author: Cheukting

hypothesis-python/src/hypothesis/extra/ghostwriter.py

@@ -891,7 +891,11 @@ def magic(
     for thing in modules_or_functions:
         if callable(thing):
             functions.add(thing)
-            funcs: list = []
+            # class need to be added for exploration
+            if inspect.isclass(thing):
+                funcs = [thing]
+            else:
+                funcs = []
         elif isinstance(thing, types.ModuleType):
             if hasattr(thing, "__all__"):
                 funcs = [getattr(thing, name, None) for name in thing.__all__]

hypothesis-python/tests/ghostwriter/recorded/fuzz_staticmethod.txt

@@ -0,0 +1,10 @@
+# This test code was written by the `hypothesis.extra.ghostwriter` module
+# and is provided under the Creative Commons Zero public domain dedication.
+
+import test_expected_output
+from hypothesis import given, strategies as st
+
+
+@given(arg=st.integers())
+def test_fuzz_A_Class_a_staticmethod(arg):
+    test_expected_output.A_Class.a_staticmethod(arg=arg)

hypothesis-python/tests/ghostwriter/recorded/magic_class.txt

@@ -0,0 +1,20 @@
+# This test code was written by the `hypothesis.extra.ghostwriter` module
+# and is provided under the Creative Commons Zero public domain dedication.
+
+import test_expected_output
+from hypothesis import given, strategies as st
+
+
+@given()
+def test_fuzz_A_Class():
+    test_expected_output.A_Class()
+
+
+@given(arg=st.integers())
+def test_fuzz_A_Class_a_classmethod(arg):
+    test_expected_output.A_Class.a_classmethod(arg=arg)
+
+
+@given(arg=st.integers())
+def test_fuzz_A_Class_a_staticmethod(arg):
+    test_expected_output.A_Class.a_staticmethod(arg=arg)

hypothesis-python/tests/ghostwriter/recorded/xml_etree_ElementTree.txt

@@ -66,6 +66,10 @@ class A_Class:
     def a_classmethod(cls, arg: int):
         pass
 
+    @staticmethod
+    def a_staticmethod(arg: int):
+        pass
+
 
 def add(a: float, b: float) -> float:
     return a + b
@@ -87,6 +91,7 @@ def divide(a: int, b: int) -> float:
         ("fuzz_sorted", ghostwriter.fuzz(sorted)),
         ("fuzz_with_docstring", ghostwriter.fuzz(with_docstring)),
         ("fuzz_classmethod", ghostwriter.fuzz(A_Class.a_classmethod)),
+        ("fuzz_staticmethod", ghostwriter.fuzz(A_Class.a_staticmethod)),
         ("fuzz_ufunc", ghostwriter.fuzz(numpy.add)),
         ("magic_gufunc", ghostwriter.magic(numpy.matmul)),
         ("magic_base64_roundtrip", ghostwriter.magic(base64.b64encode)),
@@ -177,6 +182,7 @@ def divide(a: int, b: int) -> float:
                 style="unittest",
             ),
         ),
+        ("magic_class", ghostwriter.magic(A_Class)),
         pytest.param(
             ("magic_builtins", ghostwriter.magic(builtins)),
             marks=[
@@ -186,15 +192,6 @@ def divide(a: int, b: int) -> float:
                 )
             ],
         ),
-        pytest.param(
-            ("xml_etree_ElementTree", ghostwriter.magic(xml.etree.ElementTree)),
-            marks=[
-                pytest.mark.skipif(
-                    sys.version_info[:2] not in [(3, 9), (3, 10)],
-                    reason="xml.etree.ElementTree.indent new in version 3.9",
-                )
-            ],
-        ),
     ],
     ids=lambda x: x[0],
 )