Provide fakesleep_context contextmanager.

README.rst

@@ -16,6 +16,14 @@ actually sleep. It also guarrantees "correct" results for calls to
 
 To install, simply call `fakesleep.monkey_patch()`. To restore the real versions, call `fakesleep.monkey_restore()`.
 
+A context manager is available that runs the monkeypatching and restoring for you:
+
+>>> with fakesleep.fakesleep_context():
+...    sleep(10)
+...    # no time has has actually elapsed.
+
+
+
 :author: Pete Fein <pete@wearpants.org>
 :license: BSD
 :versions: Python 2.7 & 3

fakesleep.py

@@ -1,5 +1,6 @@
 """provides a fake sleep() for use in tests"""
 
+import contextlib
 import time as time_mod
 
 originals = {name: getattr(time_mod, name) for name in
@@ -56,4 +57,17 @@ def monkey_restore():
         setattr(time_mod, k, v)
 
     global epoch
-    epoch = None
+    epoch = None
+
+@contextlib.contextmanager
+def fakesleep_context():
+    """context manager to temporarily use fakesleep.
+
+    >>> with fakesleep_context():
+    ...     sleep(10)
+    ...        
+    """
+    fakesleep.monkey_patch()
+    yield
+    fakesleep.monkey_restore()
+