diff options
author | Peter Fong <k2fong@hotmail.com> | 2017-04-11 15:44:38 -0700 |
---|---|---|
committer | Peter Fong <k2fong@hotmail.com> | 2017-04-11 15:44:38 -0700 |
commit | 5c94fb6e54bc4200966c10f944a8c28b114ebe01 (patch) | |
tree | 5729d07519caef08412f339532af3b79e72e3f2b | |
parent | f51f52daf1fc511dd3650abe454f7f1502230007 (diff) | |
download | timeout-decorator-5c94fb6e54bc4200966c10f944a8c28b114ebe01.tar.gz |
Added unit test to cover class method case
-rw-r--r-- | tests/test_timeout_decorator.py | 9 |
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(): |