diff options
| author | Sergey M․ <dstftw@gmail.com> | 2016-05-09 21:58:38 +0600 |
|---|---|---|
| committer | Sergey M․ <dstftw@gmail.com> | 2016-05-09 21:58:38 +0600 |
| commit | 129263875403841da485ac74b09960d862d23f63 (patch) | |
| tree | 8dcb265903cff8f9557ba4b6ea4f0fa3ad91d6c1 | |
| parent | fe40f9eef2483748ed83c9749f35220143d8cc9b (diff) | |
| download | youtube-dl-129263875403841da485ac74b09960d862d23f63.tar.gz youtube-dl-129263875403841da485ac74b09960d862d23f63.tar.xz youtube-dl-129263875403841da485ac74b09960d862d23f63.zip | |
[test_compat] Use compat_setenv
| -rw-r--r-- | test/test_compat.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/test/test_compat.py b/test/test_compat.py index 0d751a856..afe6bd528 100644 --- a/test/test_compat.py +++ b/test/test_compat.py @@ -27,9 +27,7 @@ from youtube_dl.compat import ( class TestCompat(unittest.TestCase): def test_compat_getenv(self): test_str = 'тест' - os.environ['YOUTUBE-DL-TEST'] = ( - test_str if sys.version_info >= (3, 0) - else test_str.encode(get_filesystem_encoding())) + compat_setenv('YOUTUBE-DL-TEST', test_str) self.assertEqual(compat_getenv('YOUTUBE-DL-TEST'), test_str) def test_compat_setenv(self): @@ -42,11 +40,9 @@ class TestCompat(unittest.TestCase): def test_compat_expanduser(self): old_home = os.environ.get('HOME') test_str = 'C:\Documents and Settings\тест\Application Data' - os.environ['HOME'] = ( - test_str if sys.version_info >= (3, 0) - else test_str.encode(get_filesystem_encoding())) + compat_setenv('HOME', test_str) self.assertEqual(compat_expanduser('~'), test_str) - os.environ['HOME'] = old_home + compat_setenv('HOME', old_home) def test_all_present(self): import youtube_dl.compat |