add tests

Author: ben-schwen

R/data.table.R

@@ -3346,7 +3346,7 @@ is_constantish = function(q, check_singleton=FALSE) {
 
 .gforce_ok = function(q, x, envir=parent.frame(2L)) {
   if (is.N(q)) return(TRUE) # For #334
-  if (!is.call(q)) return(FALSE)  # plain columns are not gforce-able since they might not aggregate (see test 104.1)
+  if (!is.call(q)) return(is.numeric(q)) # plain columns are not gforce-able since they might not aggregate (see test 104.1)
   if (q %iscall% "(") return(.gforce_ok(q[[2L]], x, envir))
 
   q1 = .get_gcall(q)

inst/tests/tests.Rraw

@@ -21236,3 +21236,7 @@ test(2345.23, levels=0:2, names(dt[, lapply(list(b, c), sum), by=a]))
 dt = data.table(a=1:4, b=1:2)
 test(2345.31, levels=0:2, dt[, .(max(a)-min(a)), by=b, verbose=TRUE], output=out)
 test(2345.32, levels=0:2, dt[, .((max(a) - min(a)) / (max(a) + min(a))), by=b, verbose=TRUE], data.table(b=1:2, V1=c(0.5, 1/3)), output=out)
+test(2345.33, levels=0:2, dt[, sum(a) / .N, b, verbose=TRUE], output=out)
+test(2345.34, levels=0:2, dt[, mean(a) * 2L + sum(a), b, verbose=TRUE], output=out)
+test(2345.35, levels=0:2, dt[, list(range=max(a)-min(a), avg=mean(a)), by=b, verbose=TRUE], output=out)
+test(2345.36, levels=0:2, dt[, .(max(a)-sqrt(min(a))), by=b, verbose=TRUE], output="GForce FALSE")