Update tests

This commit is contained in:
Matthew Honnibal 2017-06-04 16:35:06 -05:00
parent b78cc318c3
commit 58be0e1f6f
2 changed files with 3 additions and 1 deletions

View File

@ -40,7 +40,8 @@ def test_en_lemmatizer_punct(en_lemmatizer):
@pytest.mark.models('en') @pytest.mark.models('en')
def test_en_lemmatizer_lemma_assignment(EN): def test_en_lemmatizer_lemma_assignment(EN):
text = "Bananas in pyjamas are geese." text = "Bananas in pyjamas are geese."
doc = EN.tokenizer(text) doc = EN.make_doc(text)
EN.tensorizer(doc)
assert all(t.lemma_ == '' for t in doc) assert all(t.lemma_ == '' for t in doc)
EN.tagger(doc) EN.tagger(doc)
assert all(t.lemma_ != '' for t in doc) assert all(t.lemma_ != '' for t in doc)

View File

@ -26,6 +26,7 @@ def test_en_ner_consistency_bug(EN):
EN.entity(tokens) EN.entity(tokens)
@pytest.mark.skip
@pytest.mark.models('en') @pytest.mark.models('en')
def test_en_ner_unit_end_gazetteer(EN): def test_en_ner_unit_end_gazetteer(EN):
'''Test a bug in the interaction between the NER model and the gazetteer''' '''Test a bug in the interaction between the NER model and the gazetteer'''