X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/a0e07d31616102ac905c0519474d2c01db7ee392..86aea0d3a213da3be1da638b9b828e6f0ee1d59f:/test/test_cache.py diff --git a/test/test_cache.py b/test/test_cache.py index dbefad109..ce1624b68 100644 --- a/test/test_cache.py +++ b/test/test_cache.py @@ -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__':