tests/testthat/test-transform_fullsib.R

In damian-t-p/halfsibdesign: What the Package Does (One Line, Title Case)

set.seed(123)

df <- tibble::tribble(
  ~sire, ~ind, ~trait,
  1,     1,    1,
  1,     1,    2,
  1,     2,    1,
  1,     2,    2,
  2,     1,    2,
  2,     1,    1
)

df$value <- rnorm(nrow(df))

y_data <- fullsibdata(
  df,
  sire_name  = sire,
  ind_name   = ind,
  trait_name = trait,
  value_name = value
)

params <- list(
  list(mean = c(0, 0)),
  list(mean = c(0, 0))
)

balanced_data <- balance_data(y_data, params)

test_that("Balance data with conditional mean", {

  expect_s3_class(balanced_data, "fullsibdata")
  expect_equal(
    dim(balanced_data$tables[[1]]),
    dim(balanced_data$tables[[2]])
  )
  expect_true(is.balanced(balanced_data))
  
})

test_that("One-way sum-of-squares matrices computed correctly", {
  
  expect_error(ss_oneway(y_data))

  ss_mats <- ss_oneway(balanced_data)

  expect_true(
    all(sapply(
      ss_mats,
      \(X) all(identical(X, t(X)) & eigen(X)$values >= 0)
    ))
  )
})