mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-01 00:43:37 +00:00
6e2788fd8e
Add webdl types/sources. Add basic tests for get_webdl_type() in properfinder. Add properFinder _get_codec tests.
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import unittest
|
|
|
|
from sickbeard import properFinder
|
|
|
|
import sickbeard
|
|
import test_lib as test
|
|
|
|
sickbeard.SYS_ENCODING = 'UTF-8'
|
|
|
|
|
|
class ProperTests(test.SickbeardTestDBCase):
|
|
def check_webdl_type(self, cases):
|
|
for c in cases:
|
|
self.assertEqual(properFinder.get_webdl_type(*c[0]), c[1])
|
|
|
|
def check_get_codec(self, cases):
|
|
for c in cases:
|
|
self.assertEqual(properFinder._get_codec(c[0]), c[1])
|
|
|
|
def test_webdl_type(self):
|
|
self.check_webdl_type([
|
|
(('1080p.WEB.x264', 'The.Show.Name.S04E10.1080p.WEB.x264-GROUP'), 'webrip'),
|
|
(('720p.WEB-DL.DD5.1.H.264', 'The.Show.Name.720p.WEB-DL.DD5.1.H.264-GROUP'), 'webdl'),
|
|
(('1080p.AMZN.WEB-DL.DD5.1.H.264', 'The.Show.Name.1080p.AMZN.WEB-DL.DD5.1.H.264-GROUP'), 'Amazon'),
|
|
])
|
|
|
|
def test_get_codec(self):
|
|
self.check_get_codec([
|
|
('1080p.WEB.x264', '264'),
|
|
('720p.WEB.h264', '264'),
|
|
('HDTV.XviD', 'xvid'),
|
|
('720p.HEVC.x265', 'hevc'),
|
|
('1080p.HEVC.AC3', 'hevc'),
|
|
('10Bit.1080p.DD5.1.H.265', 'hevc'),
|
|
('720p.DD5.1.Widescreen.x265', 'hevc'),
|
|
])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(ProperTests)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|