tests/testthat/test01_basic.R

In RevolutionAnalytics/dplyrXdf: Tools for working with Microsoft R Server Xdf files and the dplyr package

context("Basic functionality")
mtx <- rxDataStep(mtcars, "mtx.xdf", overwrite=TRUE)

verifyData <- function(xdf, expectedClass)
{
    is.data.frame(head(xdf)) && class(xdf) == expectedClass  # test for exact class
}

x <- "mpg"
xs <- rlang::sym(x)

xnew <- "mpg2"

test_that("filter works", {
    tbl <- mtx %>% filter(mpg > 15, cyl <= 6)
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% filter((!!xs) > 15, cyl <= 6)
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% filter((!!rlang::sym(x)) > 15, cyl <= 6)
    expect_true(verifyData(tbl, "tbl_xdf"))
})

test_that("select works", {
    tbl <- mtx %>% select(mpg, cyl, drat)
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% select(!!xs, cyl, drat)
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% select(!!rlang::sym(x), cyl, drat)
    expect_true(verifyData(tbl, "tbl_xdf"))
})

test_that("subset works", {
    tbl <- mtx %>% subset(mpg > 15, c(mpg, cyl, drat))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% subset((!!xs) > 15, c(!!xs, cyl, drat))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% subset((!!rlang::sym(x)) > 15, c(!!rlang::sym(x), cyl, drat))
    expect_true(verifyData(tbl, "tbl_xdf"))
})

test_that("mutate works", {
    tbl <- mtx %>% mutate(mpg2=sin(mpg), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% mutate(mpg2=sin(!!xs), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% mutate(!!xnew := sin(!!xs), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
})

test_that("transmute works", {
    tbl <- mtx %>% transmute(mpg2=sin(mpg), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% transmute(mpg2=sin(!!xs), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% transmute(!!xnew := sin(!!xs), wt2=sqrt(wt))
    expect_true(verifyData(tbl, "tbl_xdf"))
})

test_that("output to data.frame works", {
    tbl <- mtx %>% filter(mpg > 15, cyl <= 6, .outFile=NULL)
    expect_true(is.data.frame(tbl))
    tbl <- mtx %>% select(mpg, cyl, drat, .outFile=NULL)
    expect_true(is.data.frame(tbl))
    tbl <- mtx %>% subset(mpg > 15, c(mpg, cyl, drat), .outFile=NULL)
    expect_true(is.data.frame(tbl))
    tbl <- mtx %>% mutate(mpg2=sin(mpg), wt2=sqrt(wt), .outFile=NULL)
    expect_true(is.data.frame(tbl))
    tbl <- mtx %>% transmute(mpg2=sin(mpg), wt2=sqrt(wt), .outFile=NULL)
    expect_true(is.data.frame(tbl))
})

test_that("output to xdf works", {
    tbl <- mtx %>% filter(mpg > 15, cyl <= 6, .outFile="test01.xdf")
    expect_true(verifyData(tbl, "RxXdfData"))
    tbl <- mtx %>% select(mpg, cyl, drat, .outFile="test01.xdf")
    expect_true(verifyData(tbl, "RxXdfData"))
    tbl <- mtx %>% subset(mpg > 15, c(mpg, cyl, drat), .outFile="test01.xdf")
    expect_true(verifyData(tbl, "RxXdfData"))
    tbl <- mtx %>% mutate(mpg2=sin(mpg), wt2=sqrt(wt), .outFile="test01.xdf")
    expect_true(verifyData(tbl, "RxXdfData"))
    tbl <- mtx %>% transmute(mpg2=sin(mpg), wt2=sqrt(wt), .outFile="test01.xdf")
    expect_true(verifyData(tbl, "RxXdfData"))
})

test_that(".rxArgs works", {
    tbl <- mtx %>% filter(mpg > 15, cyl <= 6, .rxArgs=list(transformFunc=function(varlst) {
        varlst$mpg2 <- sin(varlst$mpg)
        varlst
    }))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% select(mpg, cyl, drat, .rxArgs=list(transformFunc=function(varlst) {
        varlst$mpg2 <- sin(varlst$mpg)
        varlst
    }))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% subset(mpg > 15, c(mpg, cyl, drat), .rxArgs=list(transformFunc=function(varlst) {
        varlst$mpg2 <- sin(varlst$mpg)
        varlst
    }))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% mutate(wt2=sqrt(wt), .rxArgs=list(transformFunc=function(varlst) {
        varlst$mpg2 <- sin(varlst$mpg)
        varlst
    }))
    expect_true(verifyData(tbl, "tbl_xdf"))
    tbl <- mtx %>% transmute(.rxArgs=list(transformFunc=function(varlst) {
        varlst$mpg2 <- sin(varlst$mpg)
        varlst$wt2 <- sqrt(varlst$wt)
        varlst
    }))
    expect_true(verifyData(tbl, "tbl_xdf"))
})


# cleanup
file.remove("mtx.xdf", "test01.xdf")