[yt-dlp.git] / test / test_postprocessors.py

#!/usr/bin/env python

from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from yt_dlp.postprocessor import MetadataFromFieldPP, MetadataFromTitlePP


class TestMetadataFromField(unittest.TestCase):
    def test_format_to_regex(self):
        pp = MetadataFromFieldPP(None, ['title:%(title)s - %(artist)s'])
        self.assertEqual(pp._data[0]['regex'], r'(?P<title>[^\r\n]+)\ \-\ (?P<artist>[^\r\n]+)')


class TestMetadataFromTitle(unittest.TestCase):
    def test_format_to_regex(self):
        pp = MetadataFromTitlePP(None, '%(title)s - %(artist)s')
        self.assertEqual(pp._titleregex, r'(?P<title>[^\r\n]+)\ \-\ (?P<artist>[^\r\n]+)')
Commit	Line	Data
88cf6fb3 JMF	1	#!/usr/bin/env python
	2
	3	from __future__ import unicode_literals
	4
	5	# Allow direct execution
	6	import os
	7	import sys
	8	import unittest
	9	sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
	10
7a5c1cfe	11	from yt_dlp.postprocessor import MetadataFromFieldPP, MetadataFromTitlePP
5bfa4862	12
	13
	14	class TestMetadataFromField(unittest.TestCase):
	15	def test_format_to_regex(self):
	16	pp = MetadataFromFieldPP(None, ['title:%(title)s - %(artist)s'])
	17	self.assertEqual(pp._data[0]['regex'], r'(?P<title>[^\r\n]+)\ \-\ (?P<artist>[^\r\n]+)')
88cf6fb3 JMF	18
	19
	20	class TestMetadataFromTitle(unittest.TestCase):
	21	def test_format_to_regex(self):
	22	pp = MetadataFromTitlePP(None, '%(title)s - %(artist)s')
5bfa4862	23	self.assertEqual(pp._titleregex, r'(?P<title>[^\r\n]+)\ \-\ (?P<artist>[^\r\n]+)')