This commit is contained in:
Georg Brandl
2011-05-15 11:15:20 +02:00
parent 4fb8139f36
commit 3c2662580a
3 changed files with 8 additions and 10 deletions

View File

@@ -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()