소스 검색

[test_utils] Fix compat_getenv and compat_expanduser tests on python 3.x

Jaime Marquínez Ferrándiz 11 년 전
부모
커밋
8abec2c8bb
1개의 변경된 파일4개의 추가작업 그리고 2개의 파일을 삭제
  1. 4 2
      test/test_utils.py

+ 4 - 2
test/test_utils.py

@@ -360,12 +360,14 @@ class TestUtil(unittest.TestCase):
 
 
     def test_compat_getenv(self):
     def test_compat_getenv(self):
         test_str = 'тест'
         test_str = 'тест'
-        os.environ['YOUTUBE-DL-TEST'] = test_str.encode(get_filesystem_encoding())
+        os.environ['YOUTUBE-DL-TEST'] = (test_str if sys.version_info >= (3, 0)
+            else test_str.encode(get_filesystem_encoding()))
         self.assertEqual(compat_getenv('YOUTUBE-DL-TEST'), test_str)
         self.assertEqual(compat_getenv('YOUTUBE-DL-TEST'), test_str)
 
 
     def test_compat_expanduser(self):
     def test_compat_expanduser(self):
         test_str = 'C:\Documents and Settings\тест\Application Data'
         test_str = 'C:\Documents and Settings\тест\Application Data'
-        os.environ['HOME'] = test_str.encode(get_filesystem_encoding())
+        os.environ['HOME'] = (test_str if sys.version_info >= (3, 0)
+            else test_str.encode(get_filesystem_encoding()))
         self.assertEqual(compat_expanduser('~'), test_str)
         self.assertEqual(compat_expanduser('~'), test_str)
 
 
 if __name__ == '__main__':
 if __name__ == '__main__':