library(testthat)
library(lavaan)
library(semfindr)

# An SEM model with latent factors
# Labelled: Some are labelled
# User-defined parameters: At least one

mod <-
'
f1 =~ x1 + x2 + x3
f2 =~ x4 + x5 + x6
f3 =~ x7 + x8 + x9
f3 ~  a * f1 + b * f2
ab := a * b
'

dat <- sem_dat

dat0 <- dat[1:50, ]
fit0 <- lavaan::sem(mod, dat0, meanstructure = TRUE)
fit0_15 <- lavaan::sem(mod, dat0[-15, ], meanstructure = TRUE)

rerun_out <- lavaan_rerun(fit0, parallel = FALSE)
rerun_15 <- rerun_out$rerun[[15]]

est0 <- lavaan::parameterEstimates(fit0, standardized = TRUE)

est_change_rerun_test1 <- est_change(rerun_out,
                                c("~"))
est_change_rerun_test2 <- est_change(rerun_out,
                                c("~~"))
est_change_rerun_test3 <- est_change(rerun_out,
                                c("=~"))
# est_change_rerun_test4 <- est_change(rerun_out,
#                                 c(":="))
est_change_rerun_test5 <- est_change(rerun_out,
                                c("=~", "~~"))
est_change_rerun_test6 <- est_change(rerun_out,
                                c("f3 ~ f2", "=~", "~~"))
est_change_rerun_test7 <- est_change(rerun_out,
                                c("~1"))

parameters_names <- paste0(est0$lhs, est0$op, est0$rhs)
# Use label if available
parameters_names[est0$label != ""] <- est0$label[est0$label != ""]

parameters_names_paths <- parameters_names[(est0$op == "~") & !is.na(est0$z)]
parameters_names_cov <- parameters_names[(est0$op == "~~") & !is.na(est0$z)]
parameters_names_loads <- parameters_names[(est0$op == "=~") & !is.na(est0$z)]
parameters_names_def <- parameters_names[(est0$op == ":=") & !is.na(est0$z)]
parameters_names_int <- parameters_names[(est0$op == "~1") & !is.na(est0$z)]

test_that("Parameter selected by operators", {
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test1)),
        sort(c("gcd", parameters_names_paths))
      )
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test2)),
        sort(c("gcd", parameters_names_cov))
      )
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test3)),
        sort(c("gcd", parameters_names_loads))
      )
    # expect_equal(ignore_attr = TRUE,
    #     sort(colnames(est_change_rerun_test4)),
    #     sort(parameters_names_def)
    #   )
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test5)),
        sort(c(c("gcd",
               parameters_names_loads,
               parameters_names_cov)))
      )
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test6)),
        sort(c("gcd",
               parameters_names_loads,
               parameters_names_cov,
               "b"))
      )
    expect_equal(ignore_attr = TRUE,
        sort(colnames(est_change_rerun_test7)),
        sort(c("gcd", parameters_names_int))
      )
  })