add tests

Author: Enchufa2

inst/include/Rcpp/traits/result_of.h

@@ -4,7 +4,7 @@
 // result_of.h: Rcpp R/C++ interface class library -- traits to help wrap
 //
 // Copyright (C) 2010 - 2024 Dirk Eddelbuettel and Romain Francois
-// Copyright (C) 2025        Dirk Eddelbuettel, Romain Francois, Iñaki Ucar
+// Copyright (C) 2025        Dirk Eddelbuettel, Romain Francois and Iñaki Ucar
 //
 // This file is part of Rcpp.
 //

inst/tinytest/cpp/sugar.cpp

@@ -2,6 +2,7 @@
 // sugar.cpp: Rcpp R/C++ interface class library -- sugar unit tests
 //
 // Copyright (C) 2012 - 2025  Dirk Eddelbuettel and Romain Francois
+// Copyright (C) 2025         Dirk Eddelbuettel, Romain Francois and Iñaki Ucar
 //
 // This file is part of Rcpp.
 //
@@ -22,7 +23,6 @@
 using namespace Rcpp ;
 
 template <typename T>
-
 class square : public std::function<T(T)> {
 public:
 	T operator()( T t) const { return t*t ; }
@@ -222,11 +222,58 @@ NumericVector runit_na_omit( NumericVector xx ){
 }
 
 // [[Rcpp::export]]
-List runit_lapply( IntegerVector xx){
+List runit_lapply( NumericVector xx ){
+    List res = lapply( xx, square<double>() );
+    return res ;
+}
+
+// [[Rcpp::export]]
+List runit_lapply_rawfun( NumericVector xx){
+    List res = lapply( xx, raw_square );
+    return res ;
+}
+
+// [[Rcpp::export]]
+List runit_lapply_lambda(NumericVector xx){
+    List res = lapply(xx, [](double x) { return x*x; });
+    return res;
+}
+
+// [[Rcpp::export]]
+List runit_lapply_seq( IntegerVector xx){
     List res = lapply( xx, seq_len );
     return res ;
 }
 
+// [[Rcpp::export]]
+NumericVector runit_mapply2(NumericVector xx, NumericVector yy){
+    NumericVector res = mapply(xx, yy, std::plus<double>());
+    return res;
+}
+
+// [[Rcpp::export]]
+NumericVector runit_mapply2_lambda(NumericVector xx, NumericVector yy){
+    NumericVector res = mapply(xx, yy, [](double x, double y) { return x+y; });
+    return res;
+}
+
+// [[Rcpp::export]]
+NumericVector runit_mapply3_lambda(NumericVector xx, NumericVector yy, NumericVector zz){
+    NumericVector res = mapply(xx, yy, zz, [](double x, double y, double z) { return x+y+z; });
+    return res;
+}
+
+// [[Rcpp::export]]
+LogicalVector runit_mapply2_logical(NumericVector xx, NumericVector yy){
+    return all(mapply(xx, yy, std::plus<double>()) < 100.0);
+}
+
+// [[Rcpp::export]]
+List runit_mapply2_list(IntegerVector xx, IntegerVector yy){
+    List res = mapply(xx, yy, seq);
+    return res ;
+}
+
 // [[Rcpp::export]]
 List runit_minus( IntegerVector xx ){
     return List::create(
@@ -327,6 +374,12 @@ NumericVector runit_sapply_rawfun( NumericVector xx){
     return res ;
 }
 
+// [[Rcpp::export]]
+NumericVector runit_sapply_lambda(NumericVector xx){
+    NumericVector res = sapply(xx, [](double x) { return x*x; });
+    return res;
+}
+
 // [[Rcpp::export]]
 LogicalVector runit_sapply_square( NumericVector xx){
     return all( sapply( xx * xx , square<double>() ) < 10.0 );
@@ -444,6 +497,12 @@ NumericMatrix runit_outer( NumericVector xx, NumericVector yy){
     return m ;
 }
 
+// [[Rcpp::export]]
+NumericMatrix runit_outer_lambda(NumericVector xx, NumericVector yy){
+    NumericMatrix m = outer(xx, yy, [](double x, double y) { return x + y; });
+    return m ;
+}
+
 // [[Rcpp::export]]
 List runit_row( NumericMatrix xx ){
     return List::create(

inst/tinytest/test_sugar.R

@@ -1,5 +1,6 @@
 
-##  Copyright (C) 2010 - 2023  Dirk Eddelbuettel and Romain Francois
+##  Copyright (C) 2010 - 2024  Dirk Eddelbuettel and Romain Francois
+##  Copyright (C) 2025         Dirk Eddelbuettel, Romain Francois and Iñaki Ucar
 ##
 ##  This file is part of Rcpp.
 ##
@@ -302,9 +303,49 @@ expect_equal( fx( c(1:10) ), fx( c(1:10) ) )
 
 #    test.sugar.lapply <- function( ){
 fx <- runit_lapply
+expect_equal( fx( 1:10 ), as.list((1:10)^2) )
+
+
+#    test.sugar.lapply.rawfun <- function( ){
+fx <- runit_lapply_rawfun
+expect_equal( fx( 1:10 ), as.list((1:10)^2) )
+
+
+#    test.sugar.lapply.lambda <- function( ){
+fx <- runit_lapply_lambda
+expect_equal( fx( 1:10 ), as.list((1:10)^2) )
+
+
+#    test.sugar.lapply.seq <- function( ){
+fx <- runit_lapply_seq
 expect_equal( fx( 1:10 ), lapply( 1:10, seq_len ) )
 
 
+#    test.sugar.mapply2 <- function( ){
+fx <- runit_mapply2
+expect_equal( fx(1:10, 1:10) , 1:10+1:10 )
+
+
+#    test.sugar.mapply2.lambda <- function( ){
+fx <- runit_mapply2_lambda
+expect_equal( fx(1:10, 1:10) , 1:10+1:10 )
+
+
+#    test.sugar.mapply3.lambda <- function( ){
+fx <- runit_mapply3_lambda
+expect_equal( fx(1:10, 1:10, 1:10) , 1:10+1:10+1:10 )
+
+
+#    test.sugar.mapply2.logical <- function( ){
+fx <- runit_mapply2_logical
+expect_true( fx(1:10, 1:10)  )
+
+
+#    test.sugar.mapply2.list <- function( ){
+fx <- runit_mapply2_list
+expect_equal( fx(1:10, 1:10*2) , mapply(seq, 1:10, 1:10*2) )
+
+
 #    test.sugar.minus <- function( ){
 fx <- runit_minus
 expect_equal(fx(1:10) ,
@@ -380,6 +421,11 @@ fx <- runit_sapply_rawfun
 expect_equal( fx(1:10) , (1:10)^2 )
 
 
+#    test.sugar.sapply.lambda <- function( ){
+fx <- runit_sapply_lambda
+expect_equal( fx(1:10) , (1:10)^2 )
+
+
 #    test.sugar.sapply.square <- function( ){
 fx <- runit_sapply_square
 expect_true( ! fx(1:10)  )
@@ -492,6 +538,13 @@ y <- 1:5
 expect_equal( fx(x,y) , outer(x,y,"+") )
 
 
+#    test.sugar.matrix.outer.lambda <- function( ){
+fx <- runit_outer_lambda
+x <- 1:2
+y <- 1:5
+expect_equal( fx(x,y) , outer(x,y,"+") )
+
+
 #    test.sugar.matrix.row <- function( ){
 fx <- runit_row
 m <- matrix( 1:16, nc = 4 )