skip_if_not_installed("lme4")

test_that("find_weights", {
  data(mtcars)
  set.seed(123)
  mtcars$weight <- abs(rnorm(nrow(mtcars), 1, 0.3))
  m <- lm(mpg ~ wt + cyl + vs, data = mtcars, weights = weight)
  expect_identical(find_weights(m), "weight")
})

test_that("find_weights", {
  data(iris)
  set.seed(123)
  iris$wgt <- abs(rnorm(nrow(iris), 1, 0.3))
  m <- lme4::lmer(Sepal.Width ~ Sepal.Length + (1 | Species), data = iris, weights = wgt)
  expect_identical(find_weights(m), "wgt")
})


skip_if_not_installed("nlme")

data(Orthodont, package = "nlme")
Orthodont$w <- abs(rnorm(nrow(Orthodont)))

m1 <- nlme::lme(
  distance ~ age,
  data = Orthodont,
  random = ~ 1 | Subject,
  weights = nlme::varIdent(form = ~ 1 | Sex)
)

m2 <- nlme::lme(
  distance ~ age,
  data = Orthodont,
  random = ~ 1 | Subject
)

m3 <- nlme::lme(
  distance ~ age,
  data = Orthodont,
  random = ~ 1 | Subject,
  weights = ~w
)

test_that("find_weights", {
  expect_identical(find_weights(m1), "Sex")
  expect_null(find_weights(m2))
  expect_identical(find_weights(m3), "w")
})