tests/testthat/test_get_horizontal_line_data.r

In simDAG: Simulate Data from a DAG and Associated Node Information

test_that("one node", {

  expected <- data.frame(x=10,
                         y=c(4, 2),
                         label=c("Box 1", "A"),
                         type=c("none", "time_varying_node"),
                         ymax=c(14, 12),
                         ymin=c(-6, -8),
                         xmax=10,
                         xmin=-10,
                         xend=0,
                         yend=c(4, 2))

  data <- data.frame(x=0,
                     y=c(4, 3, 2, 1),
                     label=c("Box 1", "Box 2", "A", "Last Box"),
                     type=c("none", "none", "time_varying_node", "none"),
                     ymax=c(14, 13, 12, 11),
                     ymin=c(-6, -7, -8, -9),
                     xmax=10,
                     xmin=-10)

  out <- get_horizontal_line_data(box_data=data,
                                  box_hdist=10,
                                  box_2_text="Box 2",
                                  box_last_text="Last Box")
  rownames(out) <- NULL

  expect_equal(out, expected)
})

test_that("two nodes", {

  expected <- data.frame(x=10,
                         y=c(5, 3, 2),
                         label=c("Box 1", "A", "B"),
                         type=c("none", "time_varying_node",
                                "time_varying_node"),
                         ymax=c(15, 13, 12),
                         ymin=c(-5, -7, -8),
                         xmax=10,
                         xmin=-10,
                         xend=0,
                         yend=c(5, 3, 2))

  data <- data.frame(x=0,
                     y=c(5, 4, 3, 2, 1),
                     label=c("Box 1", "Box 2", "A", "B", "Last Box"),
                     type=c("none", "none", "time_varying_node",
                            "time_varying_node", "none"),
                     ymax=c(15, 14, 13, 12, 11),
                     ymin=c(-5, -6, -7, -8, -9),
                     xmax=10,
                     xmin=-10)

  out <- get_horizontal_line_data(box_data=data,
                                  box_hdist=10,
                                  box_2_text="Box 2",
                                  box_last_text="Last Box")
  rownames(out) <- NULL

  expect_equal(out, expected)
})