docs: Create ba decorator

2 files changed, 6 insertions, 2 deletions

diff --git a/docs/documentation.md b/docs/documentation.md
index d72c6b3..82f642d 100644
--- a/docs/documentation.md
+++ b/docs/documentation.md
@@ -709,11 +709,13 @@ In Python 3.2, ``GeneratorContextManager`` objects were enhanced with
 a ``__call__`` method, so that they can be used as decorators, like so:
 
 ```python
->>> @ba 
+>>> ba = before_after('BEFORE', 'AFTER')
+>>>
+>>> @ba
 ... def hello():
 ...     print('hello')
 ...
->>> hello() 
+>>> hello()
 BEFORE
 hello
 AFTER
diff --git a/src/tests/documentation.py b/src/tests/documentation.py
index 56c0e19..6a77614 100644
--- a/src/tests/documentation.py
+++ b/src/tests/documentation.py
@@ -547,6 +547,8 @@ In Python 3.2, ``GeneratorContextManager`` objects were enhanced with
 a ``__call__`` method, so that they can be used as decorators, like so:
 
 ```python
+>>> ba = before_after('BEFORE', 'AFTER')
+>>>
 >>> @ba # doctest: +SKIP
 ... def hello():
 ...     print('hello')