fix test

Signed-off-by: Alex Pickering <[email protected]>
hms-dbmi-cellenics · May 22, 2024 · 7ceedc4 · 7ceedc4
1 parent 9a1c83d
commit 7ceedc4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pipeline-runner/tests/testthat/test-qc-7-embed_and_cluster.R b/pipeline-runner/tests/testthat/test-qc-7-embed_and_cluster.R
@@ -319,11 +319,12 @@ test_that("format_cell_sets_object doesn't hardcore all keys as louvain (need fo
   cell_sets <- rbind(cell_sets, more_cell_sets)
 
   color_pool <- mock_color_pool(n_clusters)
-  algos <- c("louvain", "leiden")
+  algos <- c("louvain", "leiden", "other")
 
   for (algo in algos) {
     res <- format_cluster_cellsets(cell_sets, algo, color_pool)
-    expect_equal(res$key, algo)
+    expected <- ifelse(algo %in% c('louvain', 'leiden'), 'louvain', algo)
+    expect_equal(res$key, expected)
   }
 })