Allow negative subsetting of columns in cytoframe/cytoset

RGLab · Dec 11, 2019 · fd10e74 · fd10e74
1 parent de1b072
commit fd10e74
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/R/cytoframe.R b/R/cytoframe.R
@@ -427,8 +427,15 @@ setMethod("[",
             stop(msg, call.=FALSE)
         }
 
-        if(is.numeric(j)||is.integer(j))    
-          subset_cytoframe_by_cols(fr@pointer, j - 1)
+        if(is.numeric(j)||is.integer(j)){
+        	if(any(j < 0)){
+        		if(!all(j <= 0)){
+        			stop("Cannot mix positive and negative subscripts")
+        		}
+        		j <- (1:length(colnames(x)))[j]
+        	}
+        	subset_cytoframe_by_cols(fr@pointer, j - 1)
+        }    
         else
           stop("invalid j index!")
       }

diff --git a/R/cytoset.R b/R/cytoset.R
@@ -685,6 +685,14 @@ setMethod("[",
 
 	    if(missing(j))
 	      j <- NULL
+	    if(is.numeric(j)||is.integer(j)){
+	    	if(any(j < 0)){
+	    		if(!all(j <= 0)){
+	    			stop("Cannot mix positive and negative subscripts")
+	    		}
+	    		j <- (1:length(colnames(x)))[j]
+	    	}
+	    }
 	    x <- copy_view(x)
 	    subset_cytoset(x@pointer, i, j)
 	    x

diff --git a/tests/testthat/test-cytoset.R b/tests/testthat/test-cytoset.R
@@ -86,6 +86,11 @@ test_that("[", {
       expect_equal(sampleNames(cs1[-c(3,7,11)]), sampleNames(cs1)[-c(3,7,11)])
       expect_error(cs1[c(-3, 7, -11)], "Cannot mix positive and negative subscripts")
 
+      #Test negative subsetting of columns for cytoset and cytoframe
+      expect_equal(colnames(cs1[,-c(3,5)]), colnames(cs1)[-c(3,5)])
+      cf1 <- cs1[[1]]
+      expect_equal(colnames(cf1[,-c(2,4)]), colnames(cf1)[-c(2,4)])
+
     })
 
 test_that("subset", {