2021-02-14 02:38:13 +00:00
|
|
|
from spacy.tokens.doc import Doc
|
|
|
|
|
from spacy.vocab import Vocab
|
|
|
|
|
from spacy.pipeline._parser_internals.arc_eager import ArcEager
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_issue7056():
|
|
|
|
|
"""Test that the Unshift transition works properly, and doesn't cause
|
|
|
|
|
sentence segmentation errors."""
|
|
|
|
|
vocab = Vocab()
|
|
|
|
|
ae = ArcEager(
|
2021-06-28 09:48:00 +00:00
|
|
|
vocab.strings, ArcEager.get_actions(left_labels=["amod"], right_labels=["pobj"])
|
2021-02-14 02:38:13 +00:00
|
|
|
)
|
|
|
|
|
doc = Doc(vocab, words="Severe pain , after trauma".split())
|
|
|
|
|
state = ae.init_batch([doc])[0]
|
|
|
|
|
ae.apply_transition(state, "S")
|
|
|
|
|
ae.apply_transition(state, "L-amod")
|
|
|
|
|
ae.apply_transition(state, "S")
|
|
|
|
|
ae.apply_transition(state, "S")
|
|
|
|
|
ae.apply_transition(state, "S")
|
|
|
|
|
ae.apply_transition(state, "R-pobj")
|
|
|
|
|
ae.apply_transition(state, "D")
|
|
|
|
|
ae.apply_transition(state, "D")
|
|
|
|
|
ae.apply_transition(state, "D")
|
|
|
|
|
assert not state.eol()
|
