fix test

Author: rok-cesnovar

tests/testthat/test-utils.R

@@ -11,18 +11,18 @@ test_that("check_divergences() works", {
   csv_files <- c(test_path("resources", "csv", "model1-2-no-warmup.csv"))
   csv_output <- read_cmdstan_csv(csv_files)
   output <- "14 of 100 \\(14.0%\\) transitions ended with a divergence."
-  expect_message(check_divergences(csv_contents$post_warmup_sampler_diagnostics), output)
+  expect_message(check_divergences(csv_output$post_warmup_sampler_diagnostics), output)
 
   csv_files <- c(test_path("resources", "csv", "model1-2-no-warmup.csv"),
                  test_path("resources", "csv", "model1-2-no-warmup.csv"))
   csv_output <- read_cmdstan_csv(csv_files)
   output <- "28 of 200 \\(14.0%\\) transitions ended with a divergence."
-  expect_message(check_divergences(csv_contents$post_warmup_sampler_diagnostics), output)
+  expect_message(check_divergences(csv_output$post_warmup_sampler_diagnostics), output)
 
   csv_files <- c(test_path("resources", "csv", "model1-2-warmup.csv"))
   csv_output <- read_cmdstan_csv(csv_files)
   output <- "1 of 100 \\(1.0%\\) transitions ended with a divergence."
-  expect_message(check_divergences(csv_contents$post_warmup_sampler_diagnostics), output)
+  expect_message(check_divergences(csv_output$post_warmup_sampler_diagnostics), output)
 
 
   fit_wramup_no_samples <- testing_fit("logistic", method = "sample",
@@ -32,7 +32,7 @@ test_that("check_divergences() works", {
                           save_warmup = TRUE,
                           validate_csv = FALSE)
   csv_output <- read_cmdstan_csv(fit_wramup_no_samples$output_files())
-  expect_message(check_divergences(csv_contents$post_warmup_sampler_diagnostics), regexp = NA)
+  expect_message(check_divergences(csv_output$post_warmup_sampler_diagnostics), regexp = NA)
 })
 
 test_that("check_sampler_transitions_treedepth() works", {
@@ -42,8 +42,8 @@ test_that("check_sampler_transitions_treedepth() works", {
   output <- "16 of 100 \\(16.0%\\) transitions hit the maximum treedepth limit of 5 or 2\\^5-1 leapfrog steps."
   expect_message(
     check_sampler_transitions_treedepth(
-      csv_contents$post_warmup_sampler_diagnostics,
-      csv_contents$metadata),
+      csv_output$post_warmup_sampler_diagnostics,
+      csv_output$metadata),
     output
   )
 
@@ -53,8 +53,8 @@ test_that("check_sampler_transitions_treedepth() works", {
   output <- "32 of 200 \\(16.0%\\) transitions hit the maximum treedepth limit of 5 or 2\\^5-1 leapfrog steps."
   expect_message(
     check_sampler_transitions_treedepth(
-      csv_contents$post_warmup_sampler_diagnostics,
-      csv_contents$metadata),
+      csv_output$post_warmup_sampler_diagnostics,
+      csv_output$metadata),
     output
   )
 
@@ -63,8 +63,8 @@ test_that("check_sampler_transitions_treedepth() works", {
   output <- "1 of 100 \\(1.0%\\) transitions hit the maximum treedepth limit of 5 or 2\\^5-1 leapfrog steps."
   expect_message(
     check_sampler_transitions_treedepth(
-      csv_contents$post_warmup_sampler_diagnostics,
-      csv_contents$metadata),
+      csv_output$post_warmup_sampler_diagnostics,
+      csv_output$metadata),
     output
   )
 })