Printing improvements

R/progress.R

@@ -62,7 +62,7 @@ progress_bar_fancy <- function(n_chains, n_steps, show_overall,
                "{cli::pb_percent} ETA: {cli::pb_eta}")
   fmt_done <- paste(
     "{cli::col_green(cli::symbol$tick)} Sampled {cli::pb_total} steps",
-    "across {n_chains} chains in {cli::pb_elapsed}")
+    "across {n_chains} chain{?s} in {cli::pb_elapsed}")
   fmt_failed <- paste(
     "{cli::col_red(cli::symbol$cross)} Sampling stopped at {cli::pb_current}",
     "step{?s} after {cli::pb_elapsed}")

R/samples.R

@@ -26,6 +26,10 @@ print.monty_samples <- function(x, ...) {
     cli::cli_alert_info(
       "These samples can be restared with {.help monty_sample_continue}")
   }
+  if (!is.null(x$observations)) {
+    cli::cli_alert_info(
+      "These samples have associated observations")
+  }
 
   cli::cli_alert_info(
     paste('See {.help monty_sample} and {.run vignette("samples")} for more',

tests/testthat/test-samples.R

@@ -12,6 +12,46 @@ test_that("can print samples", {
 })
 
 
+test_that("print information about observations", {
+  model <- ex_simple_gamma1()
+  sampler <- monty_sampler_random_walk(vcv = diag(1) * 0.01)
+  s <- monty_sample(model, sampler, 100, 1, n_chains = 3)
+
+  res <- evaluate_promise(print(s))
+  expect_no_match(
+    res$messages,
+    "These samples have associated observations",
+    fixed = TRUE, all = FALSE)
+
+  s$observations <- list(a = 1)
+  res <- evaluate_promise(print(s))
+  expect_match(
+    res$messages,
+    "These samples have associated observations",
+    fixed = TRUE, all = FALSE)
+})
+
+
+test_that("print information about restart", {
+  model <- ex_simple_gamma1()
+  sampler <- monty_sampler_random_walk(vcv = diag(1) * 0.01)
+  s <- monty_sample(model, sampler, 100, 1, n_chains = 3)
+
+  res <- evaluate_promise(print(s))
+  expect_no_match(
+    res$messages,
+    "These samples can be restared with",
+    fixed = TRUE, all = FALSE)
+
+  s$restart <- list()
+  res <- evaluate_promise(print(s))
+  expect_match(
+    res$messages,
+    "These samples can be restared with",
+    fixed = TRUE, all = FALSE)
+})
+
+
 test_that("can convert to posterior draws types", {
   skip_if_not_installed("posterior")
   model <- ex_simple_gamma1()