spaCy/spacy/tests/parser/test_initial_actions_parse.py

import pytest


@pytest.mark.models
def test_initial(EN):
    doc = EN.tokenizer(u'I ate the pizza with anchovies.')
    EN.tagger(doc)
    with EN.parser.step_through(doc) as stepwise:
        stepwise.transition('L-nsubj')
        stepwise.transition('S')
        stepwise.transition('L-det')
    assert doc[0].head.i == 1
    assert doc[1].head.i == 1
    assert doc[2].head.i == 3
    assert doc[3].head.i == 3
    assert doc