[test/test_compat] Restore the old value of the HOME environment variable

[youtube-dl] / test / test_compat.py

diff --git a/test/test_compat.py b/test/test_compat.py
index d1a33ddc9f2b92a75f0d33671b9652a0813d8f18..1eb454e068970eb1a0d48cc3cd881e0bf71f9463 100644 (file)
--- a/test/test_compat.py
+++ b/test/test_compat.py
@@ -26,18 +26,20 @@ class TestCompat(unittest.TestCase):
         self.assertEqual(compat_getenv('YOUTUBE-DL-TEST'), test_str)
 
     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()))
         self.assertEqual(compat_expanduser('~'), test_str)
+        os.environ['HOME'] = old_home
 
     def test_all_present(self):
         import youtube_dl.compat
         all_names = youtube_dl.compat.__all__
         present_names = set(filter(
             lambda c: '_' in c and not c.startswith('_'),
-            dir(youtube_dl.compat)))
+            dir(youtube_dl.compat))) - set(['unicode_literals'])
         self.assertEqual(all_names, sorted(present_names))
 
 if __name__ == '__main__':