|
@@ -3,6 +3,7 @@ import pytest
|
|
import spacy
|
|
import spacy
|
|
from spacy import displacy
|
|
from spacy import displacy
|
|
|
|
|
|
|
|
+
|
|
from src.generators import Generator
|
|
from src.generators import Generator
|
|
from src.strategies import leaf_strategy
|
|
from src.strategies import leaf_strategy
|
|
|
|
|
|
@@ -15,10 +16,6 @@ def test_leaf_strategy(text, expected):
|
|
generator = Generator(text=text, strategy=leaf_strategy)
|
|
generator = Generator(text=text, strategy=leaf_strategy)
|
|
|
|
|
|
results = generator.suggest()
|
|
results = generator.suggest()
|
|
-
|
|
|
|
|
|
+ displacy.serve(docs=[generator.doc])
|
|
doc = generator.doc
|
|
doc = generator.doc
|
|
- for token in doc:
|
|
|
|
- print(token)
|
|
|
|
-
|
|
|
|
- print(results)
|
|
|
|
# assert result == expected
|
|
# assert result == expected
|