Add test and adjust version

kernprof.py

@@ -13,7 +13,7 @@
 
 # NOTE: This version needs to be manually maintained with the line_profiler
 # __version__ for now.
-__version__ = '4.0.0'
+__version__ = '4.0.1'
 
 # Guard the import of cProfile such that 3.x people
 # without lsprof can still use this script.

line_profiler/line_profiler.py

@@ -16,7 +16,7 @@
         f'Has it been compiled? Underlying error is ex={ex!r}'
     )
 
-__version__ = '4.0.0'
+__version__ = '4.0.1'
 
 
 def load_ipython_extension(ip):

tests/test_line_profiler.py

@@ -12,6 +12,12 @@ def g(x):
     y = yield x + 10
     yield y + 20
 
+class C:
+    @classmethod
+    def c(self, value):
+        print(value)
+        return 0
+
 
 def test_init():
     lp = LineProfiler()
@@ -86,3 +92,13 @@ def test_gen_decorator():
     with pytest.raises(StopIteration):
         next(i)
     assert profile.enable_count == 0
+
+def test_classmethod_decorator():
+    profile = LineProfiler()
+    c_wrapped = profile(C.c)
+    assert c_wrapped.__name__ == 'c'
+    assert profile.enable_count == 0
+    val = c_wrapped('test')
+    assert profile.enable_count == 0
+    assert val == C.c('test')
+    assert profile.enable_count == 0