tests/testthat/test-tree.R

In hadley/lobstr: Visualize R Data Structures with Trees

test_that("Array-like indices can be shown or hidden", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    tree(list(a = "a", "b", "c"), index_unnamed = TRUE)
  })
  expect_snapshot({
    tree(list(a = "a", "b", "c"), index_unnamed = FALSE)
  })
})

test_that("Atomic arrays have sensible defaults w/ truncation for longer than 10-elements",{
  testthat::skip_on_os("windows")

  expect_snapshot(
    tree(
      list(
        name = "vectored list",
        num_vec = 1:10,
        char_vec = letters
      )
    )
  )

  expect_snapshot(
    tree(
      list(
        name = "vectored list",
        num_vec = 1:10,
        char_vec = letters
      ),
      hide_scalar_types = FALSE
    )
  )

})

test_that("Large and multiline strings are handled gracefully", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    long_strings <- list(
      "normal string" = "first element",
      "really long string" = paste(rep(letters, 4), collapse = ""),
      "vec of long strings" = c(
        "a long\nand multi\nline string element",
        "a fine length",
        "another long\nand also multi\nline string element"
      )
    )

    # No truncation of first string
    # Really long single string is truncated and elipsesed
    # Short string inside vector with long strings is not truncated
    tree(long_strings)

    # Newline removal can be disabled
    tree(long_strings, remove_newlines = FALSE)
  })
})

test_that("Max depth and length can be enforced", {
  # This test also disables the unicode printing so it can be run on windows
  # platforms
  old_opts <- options("lobstr.fancy.tree" = FALSE)
  on.exit(options(old_opts))
  expect_snapshot({
    deep_list <- list(
      list(
        id = "b",
        val = 1,
        children = list(
          list(id = "b1",val = 2.5),
          list(
            id = "b2",
            val = 8,
            children = list(
              list(id = "b21", val = 4)
            )
          )
        )
      ),
      list(id = "a", val = 2)
    )
    tree(deep_list, max_depth = 1)
    tree(deep_list, max_depth = 2)
    tree(deep_list, max_depth = 3)

    tree(deep_list, max_length = 0)
    tree(deep_list, max_length = 2)
    tree(deep_list, max_depth = 1, max_length = 4)
  })
})

test_that("Missing values are caught and printed properly", {
  testthat::skip_on_os("windows")

  expect_snapshot(
    tree(
      list(
        "null-element" = NULL,
        "NA-element" = NA
      )
    )
  )
})

test_that("non-named elements in named list", {
  testthat::skip_on_os("windows")

  expect_snapshot(
    tree(list("a" = 1, "el w/o id"))
  )
})

test_that("Attributes are properly displayed as special children nodes", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    list_w_attrs <- structure(
      list(
        structure(
          list(id = "a", val = 2),
          level = 2,
          name = "first child"
        ),
        structure(
          list(
            id = "b",
            val = 1,
            children = list(
              list(id = "b1", val = 2.5)
            )
          ),
          level = 2,
          name = "second child",
          class = "custom-class"
        ),
        level = "1",
        name = "root"
      )
    )

    # Shows attributes
    tree(list_w_attrs, show_attributes = TRUE)

    # Hides attributes (default)
    tree(list_w_attrs, show_attributes = FALSE)
  })
})

test_that("Can optionally recurse into environments", {
  testthat::skip_on_os("windows")

  # Wrapped in a local to avoid different environment setup for code running in
  # test_that instead of interactively
  # Can't use snapshots here because environment address change on each run
  env_printing <- capture.output(
    local(
      {
        ea <- rlang::env(d = 4, e = 5)
        tree(rlang::env(ea, a = 1, b = 2, c = 3))
      },
      envir = rlang::global_env()
    )
  )

  # Seven total nodes should be printed
  expect_equal(
    length(env_printing),
    4
  )

  # Printed only the names we expected
  expect_equal(
    mean(
      grepl(
        pattern = "(environment|a|b|c):",
        env_printing
      )
    ),
    1
  )

  # Should only print two environment nodes (aka didn't escape past global env)
  expect_equal(
    sum(grepl(pattern = "<environment:", env_printing, fixed = TRUE)),
    1
  )

})

test_that("Function arguments get printed", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    tree(
      list(
        no_args = function(){},
        few_args = function(a,b,c){},
        lots_of_args = function(d,e,f,g,h,i,j,k,l,m,n,o,p){}
      )
    )
  })
})

test_that("Handles expressions", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    tree(
      list(
        a = quote(a),
        b = quote(a+1),
        c = y~mx+b
      )
    )
  })
})

test_that("Hidden lists dont cause infinite recursion", {
  testthat::skip_on_os("windows")

  expect_snapshot({
    tree(package_version("1.2.3"))
  })
})