Cleanup after pull from https://bitbucket.org/timmartin/sphinx

1 files changed, 3 insertions, 5 deletions

diff --git a/tests/util.py b/tests/util.py
index f15481e7b..bd50bec05 100644
--- a/tests/util.py
+++ b/tests/util.py
@@ -193,10 +193,8 @@ def with_app(*args, **kwargs):
 
 def gen_with_app(*args, **kwargs):
     """
-    Make a TestApp with args and kwargs, pass it to the test and clean up
-    properly. The wrapped function is assumed to be a generator and the
-    wrapper function is itself a generator that will iterate over the
-    values of the wrapped function and yield them.
+    Decorate a test generator to pass a TestApp as the first argument to the
+    test generator when it's executed.
     """
     def generator(func):
         @wraps(func)
@@ -226,7 +224,7 @@ def write_file(name, contents, encoding=None):
             contents = contents.encode('ascii')
     else:
         mode = 'w'
-    f = open(str(name), 'wb', encoding=encoding)
+    f = open(str(name), mode, encoding=encoding)
     f.write(contents)
     f.close()