You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
PaddleSpeech/third_party/python-pinyin/tests/test_others.py

42 lines
1.6 KiB

#!/usr/bin/env python3
from pypinyin import pinyin, Style, lazy_pinyin, slug
from pypinyin.seg.simpleseg import simple_seg
def test_import_all():
pinyin('啦啦啦') # noqa
pinyin('啦啦啦', Style.TONE2) # noqa
lazy_pinyin('啦啦啦') # noqa
slug('啦啦啦') # noqa
def test_simple_seg():
assert simple_seg('啦啦') == ['啦啦']
assert simple_seg('啦啦abc') == ['啦啦', 'abc']
assert simple_seg('&##啦啦abc') == ['&##', '啦啦', 'abc']
assert simple_seg('&#哦#啦啦abc') == ['&#', '', '#', '啦啦', 'abc']
assert simple_seg('哦ほ#') == ['', 'ほ#']
assert simple_seg(['啦啦']) == ['啦啦']
assert simple_seg(['啦啦', 'abc']) == ['啦啦', 'abc']
assert simple_seg('哦ほ#哪') == ['', 'ほ#', '']
assert simple_seg('哦ほ#哪#') == ['', 'ほ#', '', '#']
assert simple_seg('你好啊 --') == ['你好啊', ' --']
assert simple_seg('啊 -- ') == ['', ' -- ']
assert simple_seg('你好啊 -- 那') == ['你好啊', ' -- ', '']
assert simple_seg('啊 -- 你好那 ') == ['', ' -- ', '你好那', ' ']
assert simple_seg('a 你好啊 -- 那 ') == ['a ', '你好啊', ' -- ', '', ' ']
assert simple_seg('a啊 -- 你好那 ') == ['a', '', ' -- ', '你好那', ' ']
def test_issue_205():
assert pinyin('金融行业', Style.FIRST_LETTER)[2] == ['h']
assert pinyin('军工行业', Style.FIRST_LETTER)[2] == ['h']
assert pinyin('浦发银行', Style.FIRST_LETTER)[3] == ['h']
assert pinyin('交通银行', Style.FIRST_LETTER)[3] == ['h']
if __name__ == '__main__':
import pytest
pytest.cmdline.main()