# Setup -------------------------------------------------------------------

library(psych)

statistics <- list()

# alpha() -----------------------------------------------------------------

set.seed(42)

r4 <- sim.congeneric(loads = c(0.8, 0.7, 0.6, 0.5), low = -3, high = 3)
alpha_r4 <- alpha(r4)

r9 <- sim.hierarchical()
alpha_r9 <- alpha(r9)

two_f <- sim.item(8)
alpha_two_f <- alpha(two_f, keys = c("V3", "V4", "V5", "V6"), n.iter = 500)

cov_two <- cov(two_f)
alpha_cov_two <- alpha(cov_two, check.keys = TRUE)

alpha_two_f_check <- alpha(two_f, check.keys = TRUE)

items <- sim.congeneric(
  N = 500, short = FALSE, low = -2, high = 2,
  categorical = TRUE
)
alpha_a4 <- alpha(items$observed)

statistics <- statistics |>
  add_stats(alpha_r4) |>
  add_stats(alpha_r9) |>
  add_stats(alpha_two_f) |>
  add_stats(alpha_cov_two) |>
  add_stats(alpha_two_f_check) |>
  add_stats(alpha_a4)

alpha_r4
alpha_r9
alpha_two_f
alpha_cov_two
alpha_two_f_check
alpha_a4

# corr.test() -------------------------------------------------------------

ct <- corr.test(attitude)
cts <- corr.test(attitude[1:3], attitude[4:6])
cts_single <- corr.test(attitude[1:2], attitude[1:2])
cts_spearman <- corr.test(attitude[1:3], attitude[4:6], method = "spearman")
cts_kendall <- corr.test(attitude[1:3], attitude[4:6], method = "kendall")
sats <- corr.test(sat.act[1:3], sat.act[4:6], adjust = "none")
sats_no_ci <- corr.test(sat.act[1:3], sat.act[4:6], adjust = "none", ci = FALSE)
sats_alpha <- corr.test(sat.act[1:3], sat.act[4:6], adjust = "none", alpha = .1)

statistics <- statistics |>
  add_stats(ct) |>
  add_stats(cts) |>
  add_stats(sats) |>
  add_stats(sats_no_ci) |>
  add_stats(sats_alpha)

print(ct, short = FALSE)
cts
print(cts_kendall, short = FALSE)
sats
sats_no_ci
sats_alpha

# mardia() ----------------------------------------------------------------

set.seed(1)

x <- matrix(rnorm(1000), ncol = 10)

mardia_attitude <- mardia(attitude, plot = FALSE)
mardia_x <- mardia(x, plot = FALSE)

statistics <- statistics |>
  add_stats(mardia_attitude) |>
  add_stats(mardia_x)

mardia_attitude
mardia_x

# ICC() -------------------------------------------------------------------

sf <- matrix(
  ncol = 4, byrow = TRUE,
  c(
    9, 2, 5, 8,
    6, 1, 3, 2,
    8, 4, 6, 8,
    7, 1, 2, 6,
    10, 5, 6, 9,
    6, 2, 4, 7
  )
)
colnames(sf) <- paste("J", 1:4, sep = "")
rownames(sf) <- paste("S", 1:6, sep = "")

icc <- ICC(sf, lmer = FALSE, alpha = .2)

statistics <- add_stats(statistics, icc)

icc

# tidy_stats_to_data_frame() ----------------------------------------------

df <- tidy_stats_to_data_frame(statistics)

# Save stats --------------------------------------------------------------

write_test_stats(statistics, "tests/data/psych.json")

# Cleanup -----------------------------------------------------------------

rm(
  statistics, df, alpha_r4, r4, r9, two_f, items, alpha_r9, alpha_two_f,
  alpha_cov_two, alpha_two_f_check, alpha_a4, ct, cts, cts_kendall, sats,
  sats_no_ci, sats_alpha, x, mardia_attitude, mardia_x, sf, icc
)