From de6e29ae8ff0cd1bd91bec6e84119c6204837b7b Mon Sep 17 00:00:00 2001 From: Peter Baumgartner <5107405+pmbaumgartner@users.noreply.github.com> Date: Tue, 24 Jan 2023 13:34:31 -0500 Subject: [PATCH] formatting --- spacy/tests/test_cli.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/spacy/tests/test_cli.py b/spacy/tests/test_cli.py index ba7efc704..3001cfc39 100644 --- a/spacy/tests/test_cli.py +++ b/spacy/tests/test_cli.py @@ -1208,11 +1208,12 @@ def test_walk_directory(): assert (len(walk_directory(d, suffix="conll"))) == 3 assert (len(walk_directory(d, suffix="pdf"))) == 0 + def test_debug_data_trainable_lemmatizer_basic(): examples = [ ("She likes green eggs", {"lemmas": ["she", "like", "green", "egg"]}), ("Eat blue ham", {"lemmas": ["eat", "blue", "ham"]}), - ] + ] nlp = Language() train_examples = [] for t in examples: @@ -1223,6 +1224,7 @@ def test_debug_data_trainable_lemmatizer_basic(): # this results in 4 trees assert len(data["lemmatizer_trees"]) == 4 + def test_debug_data_trainable_lemmatizer_partial(): partial_examples = [ # partial annotation @@ -1244,6 +1246,7 @@ def test_debug_data_trainable_lemmatizer_partial(): data = _compile_gold(train_examples, ["trainable_lemmatizer"], nlp, True) assert data["partial_lemma_annotations"] == 2 + def test_debug_data_trainable_lemmatizer_low_cardinality(): low_cardinality_examples = [ ("She likes green eggs", {"lemmas": ["no", "no", "no", "no"]}), @@ -1257,6 +1260,7 @@ def test_debug_data_trainable_lemmatizer_low_cardinality(): data = _compile_gold(train_examples, ["trainable_lemmatizer"], nlp, True) assert data["n_low_cardinality_lemmas"] == 2 + def test_debug_data_trainable_lemmatizer_not_annotated(): unannotated_examples = [ ("She likes green eggs", {}),