youtube-dl

commit 129263875403841da485ac74b09960d862d23f63
parent fe40f9eef2483748ed83c9749f35220143d8cc9b
Author: Sergey M․ <dstftw@gmail.com>
Date:   Mon,  9 May 2016 21:58:38 +0600

[test_compat] Use compat_setenv

Diffstat:
M
test/test_compat.py
 | 
10
+++-------

1 file changed, 3 insertions(+), 7 deletions(-)
diff --git 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