tests/testthat/test-aa-export.R

In shaliulab/idocr: Analyze traces created in the Individual Drosophila Optogenetics Conditioner (IDOC)

test_that("pi_summary is exported and animals with less than minimum exists have a NA PI", {

  pkg_name <- testing_package()
  
  experiment_folder <- system.file(
    "extdata/toy", package = pkg_name,
    mustWork = TRUE
  )
  
  pi <- toy_pi_data()
  min_exits_required <- 5
  
  na_rows <- (pi$appetitive + pi$aversive) < min_exits_required
  pi[na_rows, "preference_index"] <- NA
  
  pi_path <- export_pi_summary(experiment_folder = experiment_folder, pi = pi)
  raw_data <- readLines(pi_path)[-1]
  
  # all saved pi in the rows where pi is NA is also NA
  expect_equal(
    raw_data[na_rows] %>%
      lapply(., function(x) strsplit(x, split = ",") %>% unlist %>% .[4]) %>%
      unlist %>%
      unique,
    "NA"
  )
  
  # none of the saved pi in the rows where the pi is not NA is NA
  expect_false(
    any(is.na(raw_data[!na_rows] %>%
      lapply(., function(x) strsplit(x, split = ",") %>% unlist %>% .[4]) %>%
      unlist
    ))
  )
})

test_that("main summary matches expectation in the saved file", {
  
  pkg_name <- testing_package()
  
  experiment_folder <- system.file(
    "extdata/toy", package = pkg_name,
    mustWork = TRUE
  )
  
  tracker_data <- toy_tracker_small()
  # include rois 10 and 11 in tracker  data
  tracker_data_01 <- toy_tracker_small()
  
  tracker_data_10 <- tracker_data_01[tracker_data_01$id == "toy|01", ]
  tracker_data_10$id <- "toy|10"
  tracker_data_10$region_id <- 10
  tracker_data_11 <- tracker_data_01[tracker_data_01$id == "toy|02", ]
  tracker_data_11$id <- "toy|11"
  tracker_data_11$region_id <- 11
  tracker_data <- rbind(
    tracker_data,
    tracker_data_10,
    tracker_data_11
  )
  
  controller_data <- toy_controller_small()
  
  summmary_data <- export_summary_new(
    experiment_folder = experiment_folder,
    output_csv = NULL,
    tracker_data = tracker_data,
    controller_data = controller_data
  )

  expect_snapshot_value(
    summmary_data, 
    style = "serialize", cran = FALSE
  )
})

test_that("export_dataset runs without issues", {
  
  pkg_name <- testing_package()
  
  experiment_folder <- system.file(
    "extdata/toy", package = pkg_name,
    mustWork = TRUE
  )
  
  tracker_data <- toy_tracker_small()
  controller_data <- toy_controller_small()
  pi_data <- toy_pi_data()
  
  analysis <- list(pi=pi_data[pi_data$region_id %in% unique(tracker_data$region_id), ])
  dataset <- list(
    tracker = tracker_data,
    controller = controller_data
  )
  
  all_exports <- export_dataset(experiment_folder = experiment_folder,
                 dataset = dataset, analysis = analysis)
  
  expect_true(file.exists(
    build_filename(experiment_folder, metadata = load_metadata(experiment_folder),
                   key = "SUMMARY")
  ))
  expect_true(file.exists(
    build_filename(experiment_folder, metadata = load_metadata(experiment_folder),
                   key = "PI")
  ))
  
  expect_snapshot_value(
    all_exports, 
    style = "serialize", cran = FALSE
  )
})