Added unit test to cover class method case

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/test_timeout_decorator.py b/tests/test_timeout_decorator.py
index eac0cb3..d1a0efd 100644
--- a/tests/test_timeout_decorator.py
+++ b/tests/test_timeout_decorator.py
@@ -20,6 +20,15 @@ def test_timeout_decorator_arg(use_signals):
         f()
 
 
+def test_timeout_class_method(use_signals):
+    class c():
+        @timeout(1, use_signals=use_signals)
+        def f(self):
+            time.sleep(2)
+    with pytest.raises(TimeoutError):
+        c().f()
+
+
 def test_timeout_kwargs(use_signals):
     @timeout(3, use_signals=use_signals)
     def f():