test: raw parser

This commit is contained in:
RanKKI
2022-06-26 18:29:48 +08:00
parent 5c766d7455
commit 41fd77c366
2 changed files with 24 additions and 0 deletions

3
auto_bangumi/test.sh Executable file
View File

@@ -0,0 +1,3 @@
#!/bin/bash
exec python -m unittest discover tests

View File

@@ -0,0 +1,21 @@
import unittest
from parser.analyser import RawParser
class TestRawParser(unittest.TestCase):
def test_raw_parser(self):
parser = RawParser()
content = "【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第二季 Komi-san wa, Komyushou Desu. S02】【22】【GB_MP4】【1920X1080】"
info = parser.analyse(content)
self.assertEqual(info.title, "Komi-san wa, Komyushou Desu.")
self.assertEqual(info.dpi, "1920X1080")
self.assertEqual(info.ep_info.number, 22)
self.assertEqual(info.season_info.number, 2)
def test_pre_process(self):
content = "【幻樱字幕组】【4月新番】"
expected_content = "[幻樱字幕组][4月新番]"
self.assertEqual(RawParser.pre_process(content), expected_content)