Fix plotFeature to shuffle cells correctly

Also make sure plots are returned in same order as datasets.
welch-lab · Jun 5, 2019 · abae731 · abae731
1 parent 07a1649
commit abae731
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 9 deletions.
diff --git a/R/liger.R b/R/liger.R
@@ -2342,17 +2342,17 @@ plotFeature <- function(object, feature, by.dataset = T, title = NULL, pt.size =
   } else {
     discrete <- TRUE
   }
+  if (by.dataset) {
+    dr_df$dataset <- object@cell.data$dataset
+  } else {
+    dr_df$dataset <- factor("single")
+  }
   if (do.shuffle) {
     set.seed(rand.seed)
     idx <- sample(1:nrow(dr_df))
     dr_df <- dr_df[idx, ]
   }
   p_list <- list()
-  if (by.dataset) {
-    dr_df$dataset <- object@cell.data$dataset
-  } else {
-    dr_df$dataset <- factor("single")
-  }
   for (sub_df in split(dr_df, f = dr_df$dataset)) {
     ggp <- ggplot(sub_df, aes(x = dr1, y = dr2, color = feature)) + geom_point(size = pt.size) 
 
@@ -2384,17 +2384,18 @@ plotFeature <- function(object, feature, by.dataset = T, title = NULL, pt.size =
       ggp <- ggp + theme(legend.position = "none")
     }
     p_list[[as.character(sub_df$dataset[1])]] <- ggp
-
-    if (!return.plots) {
-      print(ggp)
-    }
   }
+  p_list <- p_list[names(object@raw.data)]
   if (return.plots){
     if (length(p_list) == 1) {
       return(p_list[[1]])
     } else {
       return(p_list)
     }
+  } else {
+    for (plot in p_list) {
+      print(plot)
+    }
   }
 }
 

diff --git a/tests/testthat/test_post_factorization.R b/tests/testthat/test_post_factorization.R
@@ -131,6 +131,9 @@ plotfeatures_plots <- plotFeature(ligex, feature = 'nUMI', by.dataset = T, retur
 test_that("plotFeature returns correct ggplot objects", {
   expect_equal(length(plotfeatures_plots), length(ligex@raw.data))
   expect_is(plotfeatures_plots[[1]], class = c("ggplot"))
+  expect_equal(rownames(plotfeatures_plots[[1]]$data)[1:5], 
+               c("AATGCGTGGCTATG", "GAAAGATGATTTCC", "TTCCAAACTCCCAC", "CACTGAGACAGTCA",
+                 "GACGGCACACGGGA"))
 })
 
 # Tests for subsetting, object conversion

diff --git a/tests/testthat/test_preprocessing.R b/tests/testthat/test_preprocessing.R
@@ -23,6 +23,7 @@ test_that("Sparse matrices created", {
   expect_is(ligex@raw.data[[1]], "CsparseMatrix")
 })
 
+# note that seqwell data is previously normalized, so nUMI is 10000 for all cells
 test_that("cell.data created correctly", {
   expect_is(ligex@cell.data, "data.frame")
   expect_equal(rownames(ligex@cell.data)[1:10], colnames(ligex@raw.data[[1]])[1:10])