update tests

Author: nmdefries

facebook/delphiFacebook/R/responses.R

@@ -537,11 +537,13 @@ experimental_arm_assignment <- function(input_data, wave) {
 #'
 #' @param input_data data frame of responses
 #' @param county_crosswalk crosswalk mapping ZIP5 to counties
+#' @param params list containing `produce_individual_raceeth`, indicating
+#'   whether or not to issue microdata with race-ethnicity field
 #' @importFrom stringi stri_trim stri_replace_all
 #' @importFrom dplyr left_join group_by filter ungroup select rename
 #'
 #' @export
-create_complete_responses <- function(input_data, county_crosswalk)
+create_complete_responses <- function(input_data, county_crosswalk, params)
 {
   cols_to_report <- c(
     "start_dt", "end_dt", "date",

facebook/delphiFacebook/R/run.R

@@ -31,7 +31,7 @@ run_facebook <- function(params)
   msg_df("response data to aggregate", data_agg)
 
   # create "complete" data that will be shared with research partners
-  data_full <- create_complete_responses(input_data, cw_list$county)
+  data_full <- create_complete_responses(input_data, cw_list$county, params)
   data_full <- filter_complete_responses(data_full, params)
   data_full <- join_weights(data_full, params, weights = "full")
   msg_df("full data to share with research partners", data_full)

facebook/delphiFacebook/man/create_complete_responses.Rd

@@ -124,15 +124,17 @@ test_that("filter_data_for_aggregation works correctly", {
     hh_number_sick = c(0, NA, 4, -5, 55, 5, 5, 5, 3, 3, 0, 30, 1),
     hh_number_total = c(1, 4, NA, 5, 5, -5, 100, 5, 5, 1, 1, 30, 1),
     day = c("2021-01-01", "2021-01-01", "2021-01-02", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2020-01-01"),
-    date = c("2021-01-01", "2021-01-01", "2021-01-02", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2020-01-01")
+    date = c("2021-01-01", "2021-01-01", "2021-01-02", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01", "2020-01-01"),
+    wave = 12
   )
 
   expected <- tibble(
     zip5 = c("10001", "10001", "10001", "10001"),
     hh_number_sick = c(5, 3, 0, 30),
     hh_number_total = c(5, 5, 1, 30),
     day = c("2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01"),
-    date = c("2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01")
+    date = c("2021-01-01", "2021-01-01", "2021-01-01", "2021-01-01"),
+    wave = 12
   )
 
   expect_equal(filter_data_for_aggregation(input, params),