如何为返回数据框的函数编写单元测试的test

时间:2015-03-26 15:49:05

标签: r unit-testing testthat

我正在编写一个最终返回数据框的脚本。我的问题是,如果有任何关于如何使用单元测试包以确保返回的数据框是正确的良好实践。 (我是R程序员的开始,加上单元测试概念的新手)

我的脚本实际上如下所示:

# initialize data frame
df.out <- data.frame(...)

# function set
function1 <- function(x) {...}
function2 <- function(x) {...}

# do something to this data frame
df.out$new.column <- function1(df.out)

# do something else
df.out$other.new.column <- function2(df.out)

# etc ....

...我最终得到了一个包含许多新列的数据框。但是,使用单元测试来测试生成的数据帧是预期的最佳方法是什么?

到目前为止,我已经创建了检查每个函数结果的单元测试,但我想确保将所有这些一起运行产生预期的结果。我看过Hadley Wickham's page on testing,但在返回数据帧时无法看到有什么明显的事情。

我的想法是:

  • 手动创建预期数据框
  • 使用expect_that或类似的
  • 检查输出是否等于此数据框

关于在哪里寻找指导的任何想法/指示?到目前为止,我的谷歌已经让我失望了。

3 个答案:

答案 0 :(得分:11)

你的直觉似乎是正确的。根据函数的预期输出手动构造一个data.frame,然后将其与函数的输出进行比较。

# manually created data
dat <- iris[1:5, c("Species", "Sepal.Length")]

# function
myfun <- function(row, col, data) {
    data[row, col]
}

# result of applying function
outdat <- myfun(1:5, c("Species", "Sepal.Length"), iris)

# two versions of the same test
expect_true(identical(dat, outdat))
expect_identical(dat, outdat)

如果您的data.frame可能不是相同,您还可以在部分data.frame中运行测试,包括:

  • dim(outdat),检查尺寸是否正确
  • attributes(outdat)或列的属性
  • sapply(outdat, class),检查变量类
  • 变量摘要统计信息(如果适用)
  • 等等

答案 1 :(得分:1)

如果您想在运行时测试,请查看优秀的ensurer包,请参阅here。在页面底部,您可以看到如何构建可以测试数据框的模板,您可以根据需要进行详细和具体的模板制作。

答案 2 :(得分:0)

我只是使用这样的东西

d1 <- iris
d2 <- iris 
expect_that(d1, equals(d2)) # passes
d3 <- iris
d3[141,3] <- 5
expect_that(d1, equals(d3)) # fails