mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2026-07-22 17:37:16 +00:00
[ie/youtube] Fix extract_relative_time for abbreviated units (#16687)
Authored by: dialmaster
This commit is contained in:
@@ -9,6 +9,7 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
|
||||
|
||||
from yt_dlp.extractor import YoutubeIE
|
||||
from yt_dlp.extractor.youtube._base import YoutubeBaseInfoExtractor
|
||||
|
||||
|
||||
class TestYoutubeMisc(unittest.TestCase):
|
||||
@@ -21,6 +22,42 @@ class TestYoutubeMisc(unittest.TestCase):
|
||||
assertExtractId('http://www.youtube.com/watch?v=BaW_jenozKcsharePLED17F32AD9753930', 'BaW_jenozKc')
|
||||
assertExtractId('BaW_jenozKc', 'BaW_jenozKc')
|
||||
|
||||
def test_extract_relative_time(self):
|
||||
ert = YoutubeBaseInfoExtractor.extract_relative_time
|
||||
|
||||
# Abbreviated forms must equal their long-form equivalents.
|
||||
self.assertEqual(ert('5d ago'), ert('5 days ago'))
|
||||
self.assertEqual(ert('1mo ago'), ert('1 month ago'))
|
||||
self.assertEqual(ert('2mo ago'), ert('2 months ago'))
|
||||
self.assertEqual(ert('1y ago'), ert('1 year ago'))
|
||||
self.assertEqual(ert('1yr ago'), ert('1 year ago'))
|
||||
self.assertEqual(ert('3w ago'), ert('3 weeks ago'))
|
||||
self.assertEqual(ert('3wk ago'), ert('3 weeks ago'))
|
||||
|
||||
self.assertIsNotNone(ert('30s ago'))
|
||||
self.assertIsNotNone(ert('30sec ago'))
|
||||
self.assertIsNotNone(ert('10min ago'))
|
||||
self.assertIsNotNone(ert('5h ago'))
|
||||
self.assertIsNotNone(ert('5hr ago'))
|
||||
|
||||
self.assertIsNotNone(ert('today'))
|
||||
self.assertIsNotNone(ert('yesterday'))
|
||||
self.assertIsNotNone(ert('now'))
|
||||
|
||||
self.assertEqual(ert('5 days ago'), ert('5 day ago'))
|
||||
|
||||
self.assertIsNotNone(ert('streamed 6 days ago'))
|
||||
self.assertIsNotNone(ert('5 seconds ago (edited)'))
|
||||
self.assertIsNotNone(ert('updated today'))
|
||||
self.assertIsNotNone(ert('8 yr ago'))
|
||||
|
||||
self.assertIsNone(ert('not a date string'))
|
||||
self.assertIsNone(ert(''))
|
||||
|
||||
# Small safety check to prevent "drift".
|
||||
for unit in YoutubeBaseInfoExtractor._RELATIVE_TIME_UNIT_MAP:
|
||||
self.assertIsNotNone(ert(f'1 {unit} ago'), f'unit {unit!r} did not parse')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user