Function test_chinese_word_swap_masked

tests/test_transformations.py:152–166

()

def test_chinese_word_swap_masked():
    from textattack.augmentation import Augmenter
    from textattack.transformations.word_swaps.chn_transformations import (
        ChineseWordSwapMaskedLM,
    )

    augmenter = Augmenter(
        transformation=ChineseWordSwapMaskedLM(),
        pct_words_to_swap=0.1,
        transformations_per_example=5,
    )
    s = "自然语言处理。"
    augmented_text_list = augmenter.augment(s)
    augmented_s = "自然语言文字。"
    assert augmented_s or s in augmented_text_list