tests: misc

HLasse · Dec 20, 2022 · c12d80a · c12d80a
1 parent 9c7c032
commit c12d80a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/textdescriptives/components/coherence.py b/textdescriptives/components/coherence.py
@@ -5,9 +5,9 @@
 from spacy.tokens import Doc
 
 
-@Language.factory("textdescriptives.coherence")
+@Language.factory("textdescriptives/coherence")
 def create_coherence_component(nlp: Language, name: str):
-    """Allows Coherence to be added to a spaCy pipe using nlp.add_pipe("textdescriptives.coherence").
+    """Allows Coherence to be added to a spaCy pipe using nlp.add_pipe("textdescriptives/coherence").
     If the pipe does not contain a parser or sentencizer, the sentencizer component is silently added."""
     return Coherence(nlp)
 

diff --git a/textdescriptives/tests/test_coherence.py b/textdescriptives/tests/test_coherence.py
@@ -8,12 +8,12 @@
 @pytest.fixture(scope="function")
 def nlp():
     nlp = spacy.load("en_core_web_sm")
-    nlp.add_pipe("textdescriptives.coherence")
+    nlp.add_pipe("textdescriptives/coherence")
     return nlp
 
 
 def test_coherence_integration(nlp):
-    assert "textdescriptives.coherence" == nlp.pipe_names[-1]
+    assert "textdescriptives/coherence" == nlp.pipe_names[-1]
 
 
 def test_coherence(nlp):

diff --git a/textdescriptives/tests/test_quality.py b/textdescriptives/tests/test_quality.py
@@ -260,9 +260,8 @@ def test_passed_quality_check(text: str, passed: bool, nlp: spacy.Language):
 
 
 def test_quality_multi_process(nlp):
-    texts = [oliver_twist, secret_garden, flatland]
-    texts = [ftfy.fix_text(text) for text in texts]
+    texts = ["A couple of texts here, yeah yeah yeah.", "This is a second text, no repetition what so ever."]
 
-    docs = nlp.pipe(texts, n_process=3)
+    docs = nlp.pipe(texts, n_process=2)
     for doc in docs:
         assert doc._.quality