from m3tpostprocess import process_line, process_scn_line


def test_process_line(src, expected):
    processed = process_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


def test_process_scn_line(src, expected):
    processed = process_scn_line(src)
    if processed != expected:
        raise AssertionError(f"Expected: {expected}, but got: {processed}")


if __name__ == "__main__":
    test_process_line("Hello....", "Hello……")
    test_process_line("……。", "……")
    test_process_line("，。……", "……")
    test_process_line("…—~", "……")
    test_process_line("----~...", "——")
    test_process_line("~~--", "～～～～")
    test_process_line("!?!?", "？！？！")
    test_process_line("これはテストです测试。", "测试。")
    test_process_line("「地形模型。」", "「地形模型」")
    test_process_scn_line("「地形模型。%command;」", "「地形模型%command;」")
    test_process_scn_line("%cmd;25%的可能。", "%cmd;25％的可能。")