tests/testthat/test-filter.R

In crplyr: A 'dplyr' Interface for Crunch

context("'filter' on CrunchDataset")

with_mock_crunch({
    ds <- loadDataset("test ds")
    test_that("Some facts about the fixture dataset", {
        expect_identical(names(ds),
            c("birthyr", "gender", "location", "mymrset", "textVar", "starttime", "catarray"))
    })

    test_that("Basic filter method", {
        expect_identical(filter(ds, gender == "Male"),
            ds[ds$gender == "Male",])
        expect_identical(filter(ds, gender == "Male" | birthyr > 2000),
            ds[ds$gender == "Male" | ds$birthyr > 2000,])
        expect_identical(filter(ds, gender == "Male" & birthyr > 2000),
            ds[ds$gender == "Male" & ds$birthyr > 2000,])
    })

    test_that("Comma-separated filters", {
        expect_identical(filter(ds, gender == "Male", birthyr > 2000),
            ds[ds$gender == "Male" & ds$birthyr > 2000,])
    })

    test_that("Filter then select", {
        both <- ds %>%
            filter(gender == "Male") %>%
            select(mymrset, starttime)
        expect_identical(both, ds[ds$gender == "Male", c("mymrset", "starttime")])
    })
    
    test_that("no filter works", {
        expect_equal(ds, ds %>% filter())
    })
    
    test_that(".preserve gives error message", {
        expect_error(
            ds %>% filter(gender == "Male", .preserve = TRUE), 
            "not supported by CrunchDatasets"
        )
    })
    
    test_that("filter_ still works", {
        suppressWarnings(result <- filter_(ds, .dots = compat_as_lazy_dots(gender == "Male")))
            expect_identical(
                result,
                ds[ds$gender == "Male",]
            )
    })
})