add two tests, add DF conversion

Author: mikelove

R/dplyr_methods.R

@@ -404,7 +404,8 @@ mutate.SummarizedExperiment <- function(.data, ...) {
 mutate_features <- function(.data, ...) {
   feature_info <- rowData(.data) |>
     tibble::as_tibble() |>
-    dplyr::mutate(...)
+    dplyr::mutate(...) |>
+    as("DataFrame")
   rowData(.data) <- feature_info
   return(.data)
 }
@@ -418,13 +419,14 @@ mutate_features <- function(.data, ...) {
 #' @param ... extra arguments passed to dplyr::mutate
 #'
 #' @return a SummarizedExperiment with modified colData
-#' 
+#'
 #' @export
 mutate_samples <- function(.data, ...) {
-  feature_info <- colData(.data) |>
+  sample_info <- colData(.data) |>
     tibble::as_tibble() |>
-    dplyr::mutate(...)
-  colData(.data) <- feature_info
+    dplyr::mutate(...) |>
+    as("DataFrame")
+  colData(.data) <- sample_info
   return(.data)
 }
 

tests/testthat/test-dplyr_methods.R

@@ -205,3 +205,21 @@ test_that("group_split splits with mutliple arguments", {
     length() |> 
     expect_equal(4)
 })
+
+test_that("mutate features", {
+  pasilla %>%
+    mutate_features(new = 1:nrow(pasilla)) %>%
+    rowData() %>%
+    as_tibble() %>%
+    pull(new) %>%
+    expect_equal(1:nrow(pasilla))
+})
+
+test_that("mutate samples", {
+  pasilla %>%
+    mutate_samples(new = 1:ncol(pasilla)) %>%
+    colData() %>%
+    as_tibble() %>%
+    pull(new) %>%
+    expect_equal(1:ncol(pasilla))
+})