()
| 62 | |
| 63 | |
| 64 | def test_word_swap_change_name(): |
| 65 | from flair.data import Sentence |
| 66 | from flair.models import SequenceTagger |
| 67 | |
| 68 | from textattack.augmentation import Augmenter |
| 69 | from textattack.transformations.word_swaps import WordSwapChangeName |
| 70 | |
| 71 | augmenter = Augmenter(transformation=WordSwapChangeName()) |
| 72 | s = "My name is Anthony Davis." |
| 73 | s_augmented = augmenter.augment(s) |
| 74 | augmented_text = Sentence(s_augmented[0]) |
| 75 | tagger = SequenceTagger.load("flair/ner-english") |
| 76 | original_text = Sentence(s) |
| 77 | tagger.predict(original_text) |
| 78 | tagger.predict(augmented_text) |
| 79 | |
| 80 | entity_original = [] |
| 81 | entity_augmented = [] |
| 82 | |
| 83 | for entity in original_text.get_spans("ner"): |
| 84 | entity_original.append(entity.tag) |
| 85 | for entity in augmented_text.get_spans("ner"): |
| 86 | entity_augmented.append(entity.tag) |
| 87 | assert entity_original == entity_augmented |
| 88 | |
| 89 | |
| 90 | def test_word_swap_change_name_consistent(): |
nothing calls this directly
no test coverage detected