test_that("sample() works without a `.by` argument", {
  one <- mtcars |> sample(n=1)
  expect_equal(1, nrow(one))
})

test_that("sample() works with a character string `.by` argument", {
  one <- sample(mtcars, n=1, .by = "cyl")
  expect_equal(3, nrow(one))
  two <- sample(mtcars, n=1, .by = c("cyl", "vs"))
  expect_equal(5, nrow(two))
})

test_that("sample() works with unquoted names in `.by` argument", {
  one <- sample(mtcars, n=1, .by = cyl)
  expect_equal(3, nrow(one))
  two <- sample(mtcars, n=1, .by = c(cyl, vs))
  expect_equal(5, nrow(two))
})


test_that("sample() works with datasim object", {
  one <- sample(sim_01, n=3)
  expect_equal(3, nrow(one))
})