aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorSergey M․ <dstftw@gmail.com>2016-05-09 21:58:38 +0600
committerSergey M․ <dstftw@gmail.com>2016-05-09 21:58:38 +0600
commit129263875403841da485ac74b09960d862d23f63 (patch)
tree8dcb265903cff8f9557ba4b6ea4f0fa3ad91d6c1 /test
parentfe40f9eef2483748ed83c9749f35220143d8cc9b (diff)
[test_compat] Use compat_setenv
Diffstat (limited to 'test')
-rw-r--r--test/test_compat.py10
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