new unit tests for Sugar mean()

Author: eddelbuettel

inst/unitTests/cpp/sugar.cpp

@@ -2,7 +2,7 @@
 //
 // sugar.cpp: Rcpp R/C++ interface class library -- sugar unit tests
 //
-// Copyright (C) 2012 - 2013 Dirk Eddelbuettel and Romain Francois
+// Copyright (C) 2012 - 2015  Dirk Eddelbuettel and Romain Francois
 //
 // This file is part of Rcpp.
 //
@@ -643,3 +643,18 @@ List vector_vector_logical( NumericVector xx, NumericVector yy){
 			LogicalVector y6 = xx != yy;
 			return List::create(y1, y2, y3, y4, y5, y6);
 }
+
+// Additions made 1 Jan 2015
+
+// [[Rcpp::export]]
+double meanInteger(Rcpp::IntegerVector x) { return Rcpp::mean(x); }
+
+// [[Rcpp::export]]
+double meanNumeric(Rcpp::NumericVector x) { return(Rcpp::mean(x)); }
+
+// [[Rcpp::export]]
+double meanLogical(Rcpp::LogicalVector x) { return(Rcpp::mean(x)); }
+
+// [[Rcpp::export]]
+Rcomplex meanComplex(Rcpp::ComplexVector x) { return(Rcpp::mean(x)); }
+

inst/unitTests/runit.sugar.R

@@ -675,4 +675,38 @@ if (.runThisTest) {
                     "sugar vector vector operations")
     }
 
+    ## Additions made 1 Jan 2015
+    
+    test.mean.integer <- function() {
+        v1 <- seq(-100L, 100L)
+        v2 <- c(v1, NA)
+        checkEquals(mean(v1), meanInteger(v1), "mean of integer vector")
+        checkEquals(mean(v2), meanInteger(v2), "mean of integer vector with NA")
+    }
+    
+    test.mean.numeric <- function() {
+        v1 <- seq(-100, 100)
+        v2 <- c(v1, NA)
+        v3 <- c(v1, Inf)
+        checkEquals(mean(v1), meanNumeric(v1), "mean of numeric vector")
+        checkEquals(mean(v2), meanNumeric(v2), "mean of numeric vector with NA")
+        checkEquals(mean(v3), meanNumeric(v3), "mean of numeric vector with Inf")
+    }    
+
+    test.mean.complex <- function() {
+        v1 <- seq(-100, 100)  + 1.0i
+        v2 <- c(v1, NA)
+        v3 <- c(v1, Inf)
+        checkEquals(mean(v1), meanComplex(v1), "mean of complex vector")
+        checkEquals(mean(v2), meanComplex(v2), "mean of complex vector with NA")
+        checkEquals(mean(v3), meanComplex(v3), "mean of complex vector with Inf")
+    }
+
+    test.mean.logical <- function() {
+        v1 <- c(rep(TRUE, 50), rep(FALSE, 25))
+        v2 <- c(v1, NA)
+        checkEquals(mean(v1), meanLogical(v1), "mean of logical vector")
+        checkEquals(mean(v2), meanLogical(v2), "mean of logical vector with NA")
+    }
+    
 }