test: add classmethod flag tests

Author: zigai

tests/test_class.py

@@ -65,3 +65,8 @@ def test_class_inheritance():
     cls = Class(ExampleClassC)
     inherited_methods = [method for method in cls.methods if method.is_inherited]
     assert any(method.name == "inherited_method" for method in inherited_methods)
+
+
+def test_class_includes_classmethods_when_enabled():
+    cls = Class(ExampleClassC, classmethod=True)
+    assert "class_method" in [method.name for method in cls.methods]

tests/test_inspect.py

@@ -1,4 +1,4 @@
-from examples import ExampleClassA, example_function
+from examples import ExampleClassA, ExampleClassC, example_function
 
 from objinspect import Class, Function, Method, inspect
 
@@ -13,3 +13,11 @@ def test_correct_return_types():
 def test_method_of_instance():
     obj = inspect(ExampleClassA("a", 1).method_1)
     assert isinstance(obj, Method)
+
+
+def test_inspect_classmethod_flag():
+    default_obj = inspect(ExampleClassC)
+    assert "class_method" not in [method.name for method in default_obj.methods]
+
+    enabled_obj = inspect(ExampleClassC, classmethod=True)
+    assert "class_method" in [method.name for method in enabled_obj.methods]