# >>>>>>>>> 你可以复制下面的代码，开始运行<<<<<<<<<<<

from nlpcda.tools.randomword import Randomword
from nlpcda.tools.similarword import Similarword
from nlpcda.tools.homophone import Homophone
from nlpcda.tools.randomdeletechar import RandomDeleteChar


def test_Randomword(test_str, create_num=3, change_rate=0.1):
    smw = Randomword(create_num=create_num, change_rate=change_rate)
    return smw.replace(test_str)

def test_Similarword(test_str, create_num=3, change_rate=0.1):
    smw = Similarword(create_num=create_num, change_rate=change_rate)
    return smw.replace(test_str)

def test_Homophone(test_str, create_num=3, change_rate=0.1):
    hoe = Homophone(create_num=create_num, change_rate=change_rate)
    return hoe.replace(test_str)

def test_RandomDeleteChar(test_str, create_num=3, change_rate=0.1):
    smw = RandomDeleteChar(create_num=create_num, change_rate=change_rate)
    return smw.replace(test_str)



ts = '''这是个实体：58同城；今天是2020年3月8日11:40，天气晴朗，天气很不错，空气很好，不差；这个nlpcad包，用于方便一键数据增强，可有效增强NLP模型的泛化性能、减少波动、抵抗对抗攻击'''
rs1 = test_Randomword(ts)
rs2 = test_Similarword(ts)
rs3 = test_Homophone(ts)
rs4 = test_RandomDeleteChar(ts)

print('随机实体替换>>>>>>')
for s in rs1:
    print(s)
print('随机近义词替换>>>>>>')
for s in rs2:
    print(s)
print('随机近义字替换>>>>>>')
for s in rs3:
    print(s)

print('随机字删除>>>>>>')
for s in rs4:
    print(s)

# <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<



