Auto_Bangumi/test/test_raw_parser.py

from random import randrange

import unittest

from module.parser.analyser.raw_parser import RawParser


def test_raw_parser():
    parser = RawParser()
    content = "【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第二季 Komi-san wa, Komyushou Desu. S02】【22】【GB_MP4】【1920X1080】"
    info = parser.analyse(content)
    assert info.title_en == "Komi-san wa, Komyushou Desu."
    assert info.resolution == "1920X1080"
    assert info.episode == 22
    assert info.season == 2

    content = "[百冬练习组&LoliHouse] BanG Dream! 少女乐团派对！☆PICO FEVER！ / Garupa Pico: Fever! - 26 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕][END] [101.69 MB]"
    info = parser.analyse(content)
    assert info.group == "百冬练习组&LoliHouse"
    assert info.title_zh == "BanG Dream! 少女乐团派对！☆PICO FEVER！"
    assert info.resolution == "1080p"
    assert info.episode == 26
    assert info.season == 1

    content = "【喵萌奶茶屋】★04月新番★[夏日重现/Summer Time Rendering][11][1080p][繁日双语][招募翻译]"
    info = parser.analyse(content)
    assert info.group == "喵萌奶茶屋"
    assert info.title_en == "Summer Time Rendering"
    assert info.resolution == "1080p"
    assert info.episode == 11
    assert info.season == 1




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_en, "Komi-san wa, Komyushou Desu.")
        self.assertEqual(info.resolution, "1920X1080")
        self.assertEqual(info.episode, 22)
        self.assertEqual(info.season, 2)

        content = "[百冬练习组&LoliHouse] BanG Dream! 少女乐团派对！☆PICO FEVER！ / Garupa Pico: Fever! - 26 [WebRip 1080p HEVC-10bit AAC][简繁内封字幕][END] [101.69 MB]"
        info = parser.analyse(content)

        self.assertEqual(info.group, "百冬练习组&LoliHouse")
        self.assertEqual(info.title_zh, "BanG Dream! 少女乐团派对！☆PICO FEVER！")
        self.assertEqual(info.resolution, "1080p")
        self.assertEqual(info.episode, 26)
        self.assertEqual(info.season, 1)

        content = "【喵萌奶茶屋】★04月新番★[夏日重现/Summer Time Rendering][11][1080p][繁日双语][招募翻译]"
        info = parser.analyse(content)
        self.assertEqual(info.group, "喵萌奶茶屋")
        self.assertEqual(info.title_en, "Summer Time Rendering")
        self.assertEqual(info.resolution, "1080p")
        self.assertEqual(info.episode, 11)
        self.assertEqual(info.season, 1)

        content = "【喵萌奶茶屋】★04月新番★夏日重现/Summer Time Rendering[11][1080p][繁日双语][招募翻译]"
        info = parser.analyse(content)
        self.assertEqual(info.title_en, "Summer Time Rendering")

    def test_pre_process(self):
        content = "【幻樱字幕组】【4月新番】"
        expected_content = "[幻樱字幕组][4月新番]"
        self.assertEqual(RawParser.pre_process(content), expected_content)

    def test_get_group(self):
        content = "【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第二季 Komi-san wa, Komyushou Desu. S02】【22】【GB_MP4】【1920X1080】"

        content = RawParser.pre_process(content)
        expected_content = "幻樱字幕组"

        self.assertEqual(RawParser.get_group(content), expected_content)

    def test_find_tags(self):
        cases = [
            ("[GB_MP4] [1920X1080] [bilibili]", ["GB", "1920X1080", "bilibili"]),
            ("[GB_MP4] [1920X1080]", ["GB", "1920X1080", None]),
            ("[简_MP4] [Bilibili]", ["简", None, "Bilibili"]),
            ("[简_MP4] [Bilibili] [Web]", ["简", None, "Web"]),
            ("dfkajflkdaj dfkadjlkfa [Web]", [None, None, "Web"]),
            ("dfkajflkdaj dfkadjlkfa [Web] [Web]", [None, None, "Web"]),
        ]

        for content, expected in cases:
            ret = RawParser.find_tags(content)
            self.assertEqual(len(ret), 3)
            for i in range(3):
                self.assertEqual(ret[i], expected[i])

    def test_episode(self):
        parser = RawParser()
        for epi in range(1, 100000, 100):
            content = f"【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第一季 Komi-san wa, Komyushou Desu. S01】【{epi}】【GB_MP4】【4K】"
            info = parser.analyse(content)
            self.assertEqual(info.episode, epi)

        for epi in range(1, 100000, 100):
            content = f"[Nekomoe kissaten][Summer Time Rendering - {epi} [1080p][JPTC].mp4"
            info = parser.analyse(content)
            self.assertEqual(info.episode, epi)

    def test_season(self):
        chinese_number_arr = ["一", "二", "三", "四",
                              "五", "六", "七", "八", "九", "十", "十一", "十二"]
        parser = RawParser()
        for i in range(1, 13):
            season = str(i).zfill(2)
            content = f"【幻樱字幕组】【古见同学有交流障碍症 第{chinese_number_arr[i - 1]}季 Komi-san wa, Komyushou Desu. S{season}】[1]"
            info = parser.analyse(content)
            self.assertEqual(info.season, i)