diff options
Diffstat (limited to 'tests/unittests/helpers.py')
| -rw-r--r-- | tests/unittests/helpers.py | 5 | 
1 files changed, 4 insertions, 1 deletions
| diff --git a/tests/unittests/helpers.py b/tests/unittests/helpers.py index 7516bd02..24e1e881 100644 --- a/tests/unittests/helpers.py +++ b/tests/unittests/helpers.py @@ -288,7 +288,10 @@ def populate_dir(path, files):          os.makedirs(path)      for (name, content) in files.items():          with open(os.path.join(path, name), "wb") as fp: -            fp.write(content.encode('utf-8')) +            if isinstance(content, six.binary_type): +                fp.write(content) +            else: +                fp.write(content.encode('utf-8'))              fp.close() | 
