more unit test

Author: Qiang Kou

inst/unitTests/cpp/Subset.cpp

@@ -70,3 +70,25 @@ NumericVector subset_assign_subset2(NumericVector x) {
     y[x <= 3] = x[x > 3];
     return y;
 }
+
+// [[Rcpp::export]]
+NumericVector subset_assign_subset3(NumericVector x) {
+    NumericVector y(x.size());
+    y[x <= 3] = x[3];
+    return y;
+}
+
+// [[Rcpp::export]]
+IntegerVector subset_assign_subset4(NumericVector x) {
+    IntegerVector y(x.size());
+    y[x <= 3] = x[x <= 3];
+    return y;
+}
+
+// [[Rcpp::export]]
+NumericVector subset_assign_subset5(NumericVector x) {
+    NumericVector y(x.size());
+    y[x < 3] = x[x >= 4];
+    return y;
+}
+

inst/unitTests/runit.subset.R

@@ -78,7 +78,12 @@ if (.runThisTest) {
         checkIdentical(subset_assign_subset(1:6), c(0,0,0,4,5,6))
 
         checkIdentical(subset_assign_subset2(1:6), c(4,5,6,0,0,0))
+
+        checkIdentical(subset_assign_subset3(1:6), c(4,4,4,0,0,0))
+
+        checkIdentical(subset_assign_subset4(seq(2, 4, 0.2)), c(2L,2L,2L,2L,2L,3L,0L,0L,0L,0L,0L))
 
+        checkException(subset_assign_subset5(1:6), msg = "index error")
     }
 
 }