123456789101112131415161718192021222324 |
- import pytest
- import spacy
- from word_processor.strategies.leaf_no_preps_strategy import leaf_no_prep_strategy
- from word_processor.generators import Generator
- nlp = spacy.load("en_core_web_trf")
- @pytest.mark.parametrize('text, expected', [
- ('Regex for redacted phone numbers with extra info for PlayStation',
- ['PhoneNumberRegex', 'RedactedPhoneNumberRegex'])
- ])
- def test_leaf_no_prep_strategy(snapshot, text, expected):
- generator = Generator(text=text, strategy=leaf_no_prep_strategy)
- results = generator.suggest()
- text_results = []
- for suggestion in results:
- text_results.append([token.text for token in suggestion])
- assert text_results == snapshot
|