1#@!#!123s

: / proc / 2 / root / usr / local / lib64 / python3.6 / site-packages / spacy / tests / regression /

Filename : test_issue3830.py

back

from spacy.pipeline.pipes import DependencyParser
from spacy.vocab import Vocab


def test_issue3830_no_subtok():
    """Test that the parser doesn't have subtok label if not learn_tokens"""
    parser = DependencyParser(Vocab())
    parser.add_label("nsubj")
    assert "subtok" not in parser.labels
    parser.begin_training(lambda: [])
    assert "subtok" not in parser.labels


def test_issue3830_with_subtok():
    """Test that the parser does have subtok label if learn_tokens=True."""
    parser = DependencyParser(Vocab(), learn_tokens=True)
    parser.add_label("nsubj")
    assert "subtok" not in parser.labels
    parser.begin_training(lambda: [])
    assert "subtok" in parser.labels