mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-11 20:28:20 +03:00
0fb8d437c0
* Add test for #7035 * Update test for issue 7056 * Fix test * Fix transitions method used in testing * Fix state eol detection when rebuffer * Clean up redundant fix
28 lines
881 B
Python
28 lines
881 B
Python
import pytest
|
|
|
|
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(
|
|
vocab.strings,
|
|
ArcEager.get_actions(left_labels=["amod"], right_labels=["pobj"])
|
|
)
|
|
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()
|