tests/testthat/test07_joins.R

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

context("Joins, left source = xdf")

df1 <- data.frame(a=letters[1:20], b=1:20, c=11:30, stringsAsFactors=FALSE)
df2 <- data.frame(a=letters[7:26], d=as.character(1:20), e=11:30, stringsAsFactors=FALSE)
xdf1 <- rxDataStep(df1, "xdf1.xdf", overwrite=TRUE)
xdf2 <- rxDataStep(df2, "xdf2.xdf", overwrite=TRUE)

txt1 <- rxXdfToText(xdf1, "txt1.csv", overwrite=TRUE)
txt2 <- rxXdfToText(xdf2, "txt2.csv", overwrite=TRUE)

df1f <- data.frame(a=factor(letters[1:20]), b=1:20, c=11:30, stringsAsFactors=FALSE)
df2f <- data.frame(a=factor(letters[7:26]), d=as.character(1:20), e=11:30, stringsAsFactors=FALSE)

xdf1f <- rxDataStep(df1f, "xdf1f.xdf", overwrite=TRUE)
xdf2f <- rxDataStep(df2f, "xdf2f.xdf", overwrite=TRUE)

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


test_that("xdf to xdf joining works",
{
    expect_true(verifyData(left_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(right_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(full_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(semi_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(anti_join(xdf1, xdf2), "tbl_xdf"))
    expect_true(verifyData(union(xdf1, xdf1), "tbl_xdf"))
    expect_true(verifyData(union_all(xdf1, xdf1), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1, xdf2, by=c("b"="d")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, xdf2, by=c("b"="e")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, xdf2, by=c("c"="d")), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1f, xdf2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, xdf2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1f, xdf2), "tbl_xdf"))
})


test_that("xdf to data frame joining works",
{
    expect_true(verifyData(left_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(right_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(full_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(semi_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(anti_join(xdf1, df2), "tbl_xdf"))
    expect_true(verifyData(union(xdf1, df1), "tbl_xdf"))
    expect_true(verifyData(union_all(xdf1, df1), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1, df2, by=c("b"="d")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, df2, by=c("b"="e")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, df2, by=c("c"="d")), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1f, df2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, df2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1f, df2), "tbl_xdf"))
})


test_that("xdf to csv joining works",
{
    expect_true(verifyData(left_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(right_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(full_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(semi_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(anti_join(xdf1, txt2), "tbl_xdf"))
    expect_true(verifyData(union(xdf1, txt1), "tbl_xdf"))
    expect_true(verifyData(union_all(xdf1, txt1), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1, txt2, by=c("b"="d")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, txt2, by=c("b"="e")), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, txt2, by=c("c"="d")), "tbl_xdf"))

    expect_true(verifyData(inner_join(xdf1f, df2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1, df2f), "tbl_xdf"))
    expect_true(verifyData(inner_join(xdf1f, df2), "tbl_xdf"))
})

# clean up
file.remove(dir(pattern="\\.(csv|xdf)$"))