Handle fractional-second error rate timeouts (#187)

Author: vitalivo

funcy/flow.py

@@ -136,7 +136,7 @@ class ErrorRateExceeded(Exception):
 def limit_error_rate(fails, timeout, exception=ErrorRateExceeded):
     """If function fails to complete fails times in a row,
        calls to it will be intercepted for timeout with exception raised instead."""
-    if isinstance(timeout, int):
+    if isinstance(timeout, (int, float)):
         timeout = timedelta(seconds=timeout)
 
     def decorator(func):

tests/test_flow.py

@@ -308,3 +308,19 @@ def run():
     assert not worker.is_alive(), 'reentrant call deadlocked'
     assert not errors
     assert calls == expected
+
+
+@pytest.mark.parametrize('timeout', [0.5, timedelta(seconds=0.5)])
+def test_limit_error_rate_fractional_timeout(timeout):
+    @limit_error_rate(1, timeout)
+    def fail():
+        raise MyError
+
+    with pytest.raises(MyError):
+        fail()
+    with pytest.raises(ErrorRateExceeded):
+        fail()
+
+    fail.blocked -= timedelta(seconds=1)
+    with pytest.raises(MyError):
+        fail()