tests/testthat/test-print.R

In heavywatal/rwtl: Personal R package

test_that("printdf works", {
  expect_output(printdf(NULL), "NULL")
  expect_output(printdf(letters), "chr")
  expect_output(printdf(data.frame(x = numeric(0L))), "data.frame \\[0 x 1\\]")
  df0col = data.frame(letters)
  df0col[["letters"]] = NULL
  expect_output(printdf(df0col), "data.frame \\[26 x 0\\]")
  expect_output(printdf(VADeaths), "matrix \\[5 x 4\\]")
  expect_output(printdf(iris), "data.frame \\[150 x 5\\]")
  giris = dplyr::group_by(iris, Species)
  expect_output(printdf(giris), "Groups: Species \\[3\\]")
  expect_output(printdf(dplyr::starwars), "tbl_df")
  expect_output(printdf(ggplot2::diamonds), "<ord>")
  potus = ggplot2::presidential |> dplyr::mutate(end = as.POSIXct(end))
  expect_output(printdf(potus), "<date>\\s+<dttm>")
  expect_output(printdf(tibble::tibble(presidents)), "<ts>")
  m = matrix(seq_len(2L), nrow = 1L)
  colnames(m) = c("a", "b")
  packed = tibble::new_tibble(list(v = 1, d = data.frame(a = 2, b = 3), m = m))
  expect_output(printdf(packed), "tbl_df \\[1 x 3\\]")
  expect_output(printdf(packed), "v\\s+d\\$a\\s+d\\$b\\s+m\\[,a\\]\\s+\\[,b\\]")
})

test_that("max_print works", {
  tiris = tibble::new_tibble(iris)
  expect_output(max_print(tiris), "100")
})

test_that("print option works", {
  expect_identical(sanitize_width(0L), 10L)
  expect_identical(sanitize_width(42L), 42L)
  expect_identical(sanitize_width(99999L), 10000L)
  expect_identical(sanitize_width(NULL), getOption("width"))
  withr::with_options(generate_print_options(6L), {
    expect_identical(getOption("pillar.print_max"), 6L)
  })
})