tests/testthat/test-subsetHz.R

In ncss-tech/aqp: Algorithms for Quantitative Pedology

test_that("subsetHz works", {
  data(sp3)

  depths(sp3) <- id ~ top + bottom

  # show just horizons with 10YR hues
  res1 <- subsetHz(sp3, hue == '10YR')
  expect_equal(nrow(res1), sum(sp3$hue == '10YR'))
  
  # test retaining empty profiles with drop=FALSE
  site(sp3)$foo <- "bar"
  res2 <- subsetHz(sp3, hue == '10YR', drop = FALSE)
  expect_equal(length(res2), length(sp3))
  expect_equal(sum(is.na(res2$foo)), 2)
})