tests/testthat/test-featurise_combine_min.R

In eider: Declarative Feature Extraction from Tabular Data Records

ae2_table_path <- "../data/ae2.csv"

test_that("featurise_combine_min", {
  all_tables <- read_data(list(ae2 = ae2_table_path))
  combination <- featurise(
    all_tables,
    json_to_feature("../spec/test_combine_min.json"),
  )

  # Calculate through standard means
  # First feature: Count of 101s in diagnosis_2
  expected_feat_1 <- read_one_table(ae2_table_path) %>%
    filter(diagnosis_2 == 101) %>%
    group_by(id) %>%
    summarise(count_of_101s_in_diag2 = n()) %>%
    select(id, count_of_101s_in_diag2)
  # Second feature: Sum of all values of diagnosis_1
  # This is obviously meaningless, but it's just for testing<
  expected_feat_2 <- read_one_table(ae2_table_path) %>%
    group_by(id) %>%
    summarise(sum_of_all_diag1 = sum(diagnosis_1)) %>%
    select(id, sum_of_all_diag1)
  # Combine the two features
  expected <- full_join(expected_feat_1, expected_feat_2, by = "id") %>%
    mutate(
      count_of_101s_in_diag2 =
        tidyr::replace_na(count_of_101s_in_diag2, 0)
    ) %>%
    mutate(sum_of_all_diag1 = tidyr::replace_na(sum_of_all_diag1, 0)) %>%
    mutate(my_min = pmin(count_of_101s_in_diag2, sum_of_all_diag1)) %>%
    select(id, my_min)

  # Check the result
  expect_equal(combination$feature_table, expected)
})