tests/component-tests/test-FeatureStore.R

In data-science-competitions/Modeling-Earthquake-Damage: Richter's Predictor: Modeling Earthquake Damage

context("component test for FeatureStore")

testthat::setup({
    assign("test_env", testthat::test_env(), envir = parent.frame())
})

test_that("FeatureStore can be instantiated", {
    attach(test_env)
    expect_class(test_env$obj <- FeatureStore$new(), "FeatureStore")
})

test_that("FeatureStore has access to the project datasets", {
    attach(test_env)
    expect_class(obj$tidy_data, "data.frame")
    expect_table_has_col_names(obj$tidy_data, c(".set_source", "building_id"))
    expect_has_no_duplicates(obj$tidy_data$building_id)
})

test_that("FeatureStore has access to the treated geo features", {
    attach(test_env)
    geo_feature_names <- c(
        "geo_level_1_id_catP", "geo_level_1_id_catN", "geo_level_1_id_catD",
        "geo_level_2_id_catP", "geo_level_2_id_catN", "geo_level_2_id_catD",
        "geo_level_3_id_catP", "geo_level_3_id_catN", "geo_level_3_id_catD"
    )
    expect_class(geo_features <- obj$geo_features, "data.frame")
    expect_table_has_col_names(geo_features, c("building_id", geo_feature_names))
})

test_that("FeatureStore has access to the treated age features", {
    attach(test_env)
    age_feature_names <- c("treat_age", "treat_age_NA")
    expect_class(age_features <- obj$age_features, "data.frame")
    expect_table_has_col_names(age_features, c("building_id", age_feature_names))
})

test_that("FeatureStore has access to the underlying structures of factor variables", {
    attach(test_env)
    mfa_feature_names <- paste0("mfa_dim_", 1:3)
    expect_class(cores <- obj$mfa_features, "data.frame")
    expect_table_has_col_names(cores, c("building_id", mfa_feature_names))
})

testthat::teardown(test_env <- NULL)