Fixed issue with PicklingError causes the timeout to never be reached.

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/test_timeout_decorator.py b/tests/test_timeout_decorator.py
index 12d5389..e8e9fbf 100644
--- a/tests/test_timeout_decorator.py
+++ b/tests/test_timeout_decorator.py
@@ -48,3 +48,15 @@ def test_function_name(use_signals):
         pass
 
     assert func_name.__name__ == 'func_name'
+
+
+def test_timeout_pickle_error():
+    """Test that when a pickle error occurs a timeout error is raised."""
+    @timeout(seconds=1, use_signals=False)
+    def f():
+        time.sleep(0.1)
+        class Test(object):
+            pass
+        return Test()
+    with pytest.raises(TimeoutError):
+        f()