tests/testthat/test-auto_layout_mediation.R

In semptools: Customizing Structural Equation Modelling Plots

library(lavaan)
library(semPlot)

test_that("auto_layout_mediation", {

mod_pa <-
 'x3 ~ x1 + x2 + c3
  x4 ~ x3 + x1 + x2
  x5 ~ x4 + c1
  y ~ x5 + x3 + c2
 '

fit <- lavaan::sem(
          mod_pa,
          do.fit = FALSE
        )
dat <- simulateData(
          parameterTable(fit),
          sample.nobs = 500,
          seed = 1234
        )
fit <- lavaan::sem(
          mod_pa,
          dat
        )

m <- auto_layout_mediation(
        fit,
        exclude = c("c1", "c2", "c3")
      )


m_chk <-structure(c(NA, "x2", NA, "x1", NA, NA, "x3", NA, "x4", NA, NA,
NA, NA, "x5", NA, NA, NA, NA, "y", NA), dim = 4:5)

expect_equal(m,
             m_chk)

pm <- semPlotModel(fit) |> drop_nodes(c("c1", "c2", "c3"))
p0 <- semPaths(
          pm,
          whatLabels = "est",
          layout = m,
          DoNotPlot = TRUE
        )
# plot(p0)
expect_true(check_graph_pass_thru(p0))

mxy <- auto_layout_mediation(
        fit,
        exclude = c("c1", "c2", "c3"),
        output = "xy"
      )

expect_equal(layout_matrix_from_mxy(mxy),
             m_chk)

pm <- semPlotModel(fit) |> drop_nodes(c("c1", "c2", "c3"))
p0 <- semPaths(
          pm,
          whatLabels = "est",
          DoNotPlot = TRUE
        )

m0 <- auto_layout_mediation(
        p0,
        update_plot = FALSE
      )

expect_equal(m0,
             m_chk)


# Check special cases

m <- auto_layout_mediation(
        fit,
        exclude = c("y")
      )

pm <- semPlotModel(fit) |> drop_nodes(c("y", "c2"))
p0 <- semPaths(
          pm,
          whatLabels = "est",
          layout = m,
          exoCov = FALSE,
          DoNotPlot = TRUE
        )
expect_true(check_graph_pass_thru(p0))
# plot(p0)

m <- auto_layout_mediation(
        fit,
        v_pos = "lower"
      )

pm <- semPlotModel(fit)
p0 <- semPaths(
          pm,
          whatLabels = "est",
          layout = m,
          exoCov = FALSE,
          DoNotPlot = TRUE
        )
expect_true(check_graph_pass_thru(p0))
# plot(p0)

m <- auto_layout_mediation(
        fit,
        v_pos = "upper"
      )

pm <- semPlotModel(fit)
p0 <- semPaths(
          pm,
          whatLabels = "est",
          layout = m,
          exoCov = FALSE,
          DoNotPlot = TRUE
        )
expect_true(check_graph_pass_thru(p0))
# plot(p0)

expect_error(auto_layout_mediation(
        fit,
        exclude = c("x4")
      ))

})