tests/testthat/test_prune_nodes.R

In twang2218/pmv: Process Map Visualization

context("prune_nodes()")

test_that("prune_nodes(percentage) should be able prune nothing", {
  p <- create_pmap_graph(
    nodes = data.frame(
      name = c("a", "b", "c", "d", "e"),
      category = c("campaign", "campaign", "campaign", "sale", "sale"),
      amount = c(10, 30, 20, 40, 5),
      stringsAsFactors = FALSE
    ),
    edges = data.frame(
      from = c("a", "b", "b", "a"),
      to = c("b", "c", "d", "e"),
      amount = c(10, 30, 20, 40),
      stringsAsFactors = FALSE
    ),
    target_categories = c("sale")
  )

  for (rank in c("amount", "in_degree", "out_degree")) {
    nodes_count_before_prune <- nrow(DiagrammeR::get_node_df(p))

    # 0.1 * 4 = 0.4 ~= 0, so `prune_nodes()` should prune nothing
    p1 <- prune_nodes(p, percentage = 0.1, rank = rank)

    nodes_count_after_prune <- nrow(DiagrammeR::get_node_df(p1))
    expect_equal(nodes_count_after_prune, nodes_count_before_prune)
  }

})

test_that("prune_nodes(percentage) should be able prune half of the nodes", {
  for (rank in c("amount", "in_degree", "out_degree")) {
    p <- create_pmap_graph(
      nodes = data.frame(
        name = c("a", "b", "c", "d", "e"),
        category = c("campaign", "campaign", "campaign", "sale", "sale"),
        amount = c(10, 30, 20, 40, 5),
        stringsAsFactors = FALSE
      ),
      edges = data.frame(
        from = c("a", "b", "b", "a", "c", "d", "c"),
        to = c("b", "c", "d", "e", "d", "e", "e"),
        amount = c(10, 30, 20, 40, 50, 60, 70),
        stringsAsFactors = FALSE
      ),
      target_categories = c("sale")
    )

    nodes_count_before_prune <- DiagrammeR::count_nodes(p)

    # 0.5 * 4 = 2, so `prune_nodes()` should prune 2 nodes,
    # which should be "a => b", "b => c" according to the `amount`
    p <- prune_nodes(p, percentage = 0.5, rank = rank)

    nodes_count_after_prune <- DiagrammeR::count_nodes(p)

    expect_equal(nodes_count_after_prune, nodes_count_before_prune - round(nodes_count_before_prune * 0.5))
  }
})


test_that("prune_nodes(percentage) should be able prune all of the nodes", {

  for (rank in c("amount", "in_degree", "out_degree")) {
    p <- create_pmap_graph(
      nodes = data.frame(
        name = c("a", "b", "c", "d", "e"),
        category = c("campaign", "campaign", "campaign", "sale", "sale"),
        amount = c(10, 30, 20, 40, 5),
        stringsAsFactors = FALSE
      ),
      edges = data.frame(
        from = c("a", "b", "b", "a"),
        to = c("b", "c", "d", "e"),
        amount = c(10, 30, 20, 40),
        stringsAsFactors = FALSE
      ),
      target_categories = c("sale")
    )

    nodes_count_before_prune <- nrow(DiagrammeR::get_node_df(p))

    # 1 * 4 = 4, so `prune_nodes()` should prune all 4 nodes
    p <- prune_nodes(p, percentage = 1, rank = rank)

    nodes_count_after_prune <- nrow(DiagrammeR::get_node_df(p))
    expect_equal(nodes_count_after_prune, 0)
  }
})




test_that("prune_nodes(max) should be able prune nothing", {
  p <- create_pmap_graph(
    nodes = data.frame(
      name = c("a", "b", "c", "d", "e"),
      category = c("campaign", "campaign", "campaign", "sale", "sale"),
      amount = c(10, 30, 20, 40, 5),
      stringsAsFactors = FALSE
    ),
    edges = data.frame(
      from = c("a", "b", "b", "a"),
      to = c("b", "c", "d", "e"),
      amount = c(10, 30, 20, 40),
      stringsAsFactors = FALSE
    ),
    target_categories = c("sale")
  )

  for (rank in c("amount", "in_degree", "out_degree")) {
    nodes_count_before_prune <- nrow(DiagrammeR::get_node_df(p))

    # total 5 nodes, max = 5, so nothing should be pruned.
    p1 <- prune_nodes(p, max = 5, rank = rank)

    nodes_count_after_prune <- nrow(DiagrammeR::get_node_df(p1))
    expect_equal(nodes_count_after_prune, nodes_count_before_prune)
  }

})

test_that("prune_nodes(max) should be able prune half of the nodes", {
  for (rank in c("amount", "in_degree", "out_degree")) {
    p <- create_pmap_graph(
      nodes = data.frame(
        name = c("a", "b", "c", "d", "e"),
        category = c("campaign", "campaign", "campaign", "sale", "sale"),
        amount = c(10, 30, 20, 40, 5),
        stringsAsFactors = FALSE
      ),
      edges = data.frame(
        from = c("a", "b", "b", "a", "c", "d", "c"),
        to = c("b", "c", "d", "e", "d", "e", "e"),
        amount = c(10, 30, 20, 40, 50, 60, 70),
        stringsAsFactors = FALSE
      ),
      target_categories = c("sale")
    )

    nodes_count_before_prune <- DiagrammeR::count_nodes(p)

    # total 4 edges, `max = 2`, so 2 nodes should be pruned
    p <- prune_nodes(p, max = 2, rank = rank)

    nodes_count_after_prune <- DiagrammeR::count_nodes(p)

    expect_equal(nodes_count_after_prune, 2)
  }
})


test_that("prune_nodes(max) should be able prune all of the nodes", {

  for (rank in c("amount", "in_degree", "out_degree")) {
    p <- create_pmap_graph(
      nodes = data.frame(
        name = c("a", "b", "c", "d", "e"),
        category = c("campaign", "campaign", "campaign", "sale", "sale"),
        amount = c(10, 30, 20, 40, 5),
        stringsAsFactors = FALSE
      ),
      edges = data.frame(
        from = c("a", "b", "b", "a"),
        to = c("b", "c", "d", "e"),
        amount = c(10, 30, 20, 40),
        stringsAsFactors = FALSE
      ),
      target_categories = c("sale")
    )

    nodes_count_before_prune <- nrow(DiagrammeR::get_node_df(p))

    # total 4 nodes, `max = 0`, so all nodes should be pruned
    p <- prune_nodes(p, max = 0, rank = rank)

    nodes_count_after_prune <- nrow(DiagrammeR::get_node_df(p))
    expect_equal(nodes_count_after_prune, 0)
  }
})