tests/testthat/test_compare_model.R

In UBC-MDS/hamr: Handle All Missing Values

library(dplyr)
library(stats)
library(utils)

context("Testing compare_model")
df <- read.csv("../missing_valuesdf.csv")

meds <- c("CC","MIP")
result <- compare_model(df,'col1', meds,NA)

test <- broom::tidy(df) %>%
  select(column,mean,sd,min,median,max) %>%
  filter(column == "col1")

for(method in meds){
  df_after <- impute_missing(df,'col1',method,NA)
  name <- paste('col1_after_', method, sep="")
  b <- broom::tidy(df_after) %>%
    select(column,mean,sd,min,median,max) %>%
    filter(column == "col1")
  b$column[1] <- name
  test <- rbind(test,b)
}

test <- test[-1,]

test_that("compare_model(df,feature, method,missing_val_type),
          Test this function for a normal case", {
    expect_match(typeof(result),"list")
    expect_match(class(result),"data.frame")
    expect_equal(test,result)

    # expected errors
    expect_error(compare_model(list(1, 2, 2, NA), "V2", "MIP", "NA"),
                 "Error: the specified column name is not in the data frame")

    expect_error(compare_model(data.frame("exp" = c(1, 2, 3), "res" = c(0, 10, NaN)),
                                2,
                                "CC",
                                "NaN"),
                 "Error: column name is not applicable, expected a string instead")

    expect_error(compare_model(data.frame(exp = c(1, 2, 3), res = c(0, 10, "")),
                                "dn",
                                "CC",
                                ""),
                 "Error: the specified column name is not in the data frame")

    expect_error(compare_model(data.frame(x = c(1, 2, 3), y = c(0, 10, NaN)), "y", "multi_im", "NaN"),
                 "Error: method is not applicable")

    expect_error(compare_model(data.frame(x = c(1, 2, 3), y = c(0, 10, 0)), "y", "CC", 0),
                 "Error: missing value format is not supported, expected one of blank space, a question mark, NA and NaN")
})