fix: Update TestSlideEmbedder to include merge arg when saving embedd…

…ings
gustaveroussy · Aug 12, 2024 · 48afecc · 48afecc
1 parent 06dfc76
commit 48afecc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_emb.py b/tests/test_emb.py
@@ -250,7 +250,7 @@ def test_save_embeddings(self, format, merge, subtests):
             slide_name, _ = os.path.basename(slide_path).split(".")
             with subtests.test(msg=f"Test slide {slide_name}"):
                 slide_embedder = self.slide_embedders[i]
-                slide_embedder.save_embeddings(self.embeddings_dir, format=format)
+                slide_embedder.save_embeddings(self.embeddings_dir, format=format, merge=merge)
                 if merge:
                     assert os.path.exists(f"{self.embeddings_dir}/{slide_name}.{format}")
                     emb = load_embeddings(f"{self.embeddings_dir}/{slide_name}.{format}", format)
@@ -260,4 +260,4 @@ def test_save_embeddings(self, format, merge, subtests):
                     for suffix in ["model_based", "cell_based", "stain_based"]:
                         assert os.path.exists(f"{self.embeddings_dir}/{slide_name}_{suffix}.{format}")
                         emb_files.append(load_embeddings(f"{self.embeddings_dir}/{slide_name}_{suffix}.{format}", format))
-                    assert all([len(emb) == self.n_samples_coords for emb in emb_files])        
+                    assert all([len(emb) == self.n_samples_coords for emb in emb_files])