Sped up tests and fixed error in figures test

x-tabdeveloping · Jul 2, 2024 · 1439d86 · 1439d86
1 parent 60c3583
commit 1439d86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tests/test_figures.py b/tests/test_figures.py
@@ -26,7 +26,9 @@ def test_figures():
     )
     texts = newsgroups.data
     labels = list(np.array(newsgroups.target_names)[newsgroups.target])
-    trf = SentenceTransformer("all-MiniLM-L6-v2")
+    trf = SentenceTransformer(
+        "sentence-transformers/average_word_embeddings_glove.6B.300d"
+    )
     embeddings = np.asarray(trf.encode(texts))
     models = dict()
     models["nmf"] = make_topic_pipeline(
@@ -37,8 +39,6 @@ def test_figures():
         TfidfVectorizer(stop_words="english", max_features=8000),
         TruncatedSVD(10),
     ).fit(texts)
-    models["s3"] = SemanticSignalSeparation(10)
-    models["keynmf"] = KeyNMF(10)
     example_document = "Joe Biden takes over presidential office from Donald Trump."
     plots = {
         "group_map": partial(figures.group_map, group_labels=labels),

diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -16,8 +16,7 @@ def test_app_integration():
     newsgroups = fetch_20newsgroups(
         subset="all",
         categories=[
-            "misc.forsale",
-            "sci.med",
+            "alt.atheism",
         ],
         remove=("headers", "footers", "quotes"),
     )