]> jfr.im git - yt-dlp.git/blobdiff - test/test_cache.py
[misc] Cleanup (#9765)
[yt-dlp.git] / test / test_cache.py
index dbefad109e16d358ebd8b804936841940abffa25..ce1624b68b7ab2e3016f3290b77515de7b299409 100644 (file)
@@ -1,19 +1,17 @@
-#!/usr/bin/env python
-# coding: utf-8
-
-from __future__ import unicode_literals
-
-import shutil
+#!/usr/bin/env python3
 
 # Allow direct execution
 import os
 import sys
 import unittest
+
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 
+import shutil
+
 from test.helper import FakeYDL
-from youtube_dl.cache import Cache
+from yt_dlp.cache import Cache
 
 
 def _is_empty(d):
@@ -43,16 +41,16 @@ def test_cache(self):
         })
         c = Cache(ydl)
         obj = {'x': 1, 'y': ['รค', '\\a', True]}
-        self.assertEqual(c.load('test_cache', 'k'), None)
-        c.store('test_cache', 'k', obj)
+        self.assertEqual(c.load('test_cache', 'k.'), None)
+        c.store('test_cache', 'k.', obj)
         self.assertEqual(c.load('test_cache', 'k2'), None)
         self.assertFalse(_is_empty(self.test_dir))
-        self.assertEqual(c.load('test_cache', 'k'), obj)
+        self.assertEqual(c.load('test_cache', 'k.'), obj)
         self.assertEqual(c.load('test_cache', 'y'), None)
-        self.assertEqual(c.load('test_cache2', 'k'), None)
+        self.assertEqual(c.load('test_cache2', 'k.'), None)
         c.remove()
         self.assertFalse(os.path.exists(self.test_dir))
-        self.assertEqual(c.load('test_cache', 'k'), None)
+        self.assertEqual(c.load('test_cache', 'k.'), None)
 
 
 if __name__ == '__main__':