tests/testthat/test-plot.R

In abodein/netOmics: Multi-Omics (time-course) network-based integration and interpretation

context("plot")

graph1 <- igraph::graph_from_data_frame(list(from = c("A", "B", "A", "D", "C", "A", "C"), 
                                             to = c("B", "C", "D", "E", "D", "F", "G")), directed = FALSE)
graph1 <- set_vertex_attr(graph = graph1, name = 'type', index = c("A","B","C"),value = "1")
graph1 <- set_vertex_attr(graph = graph1, name = 'type', index = c("D","E"),value = "2")
graph1 <- set_vertex_attr(graph = graph1, name = 'type', index = c("F", "G"),value = "3")

rwr_res <- random_walk_restart(X = graph1, seed = c("A", "B", "C", "D", "E"))
rwr_res_type <- rwr_find_seeds_between_attributes(X = rwr_res, attribute = "type", k = 3)

graph1.list <- list("X" = graph1, "Y"= graph1)

rwr_res.list <- random_walk_restart(X = graph1.list, seed = c("A", "B", "C", "D", "E"))
rwr_res_type.list <- rwr_find_seeds_between_attributes(X = rwr_res.list, attribute = "type", k = 3)

summary_plot_rwr_attributes(rwr_res_type)

test_that("summary_plot_rwr_attributes fails on invalid input", {
    #X
    expect_error(summary_plot_rwr_attributes(X = c()))
    expect_error(summary_plot_rwr_attributes(X = NULL))
    # color must be a named vector/list
    expect_error(summary_plot_rwr_attributes(X = rwr_res_type, color = 3), "'color' must be a named verctor or list", fixed = TRUE)
    expect_error(summary_plot_rwr_attributes(X = rwr_res_type, color = data.frame()), "'color' must be a named verctor or list", fixed = TRUE)
    
    # seed.id 
    expect_error(summary_plot_rwr_attributes(X = rwr_res_type, seed.id = c(1)), "'seed.id' must be a charactor vector", fixed = TRUE)
    # seed.type
    expect_error(summary_plot_rwr_attributes(X = rwr_res_type, seed.type = c(1)), "'seed.type' must be a charactor vector", fixed = TRUE)
    
    # plot TRUE/FALSE -> default
})

test_that("summary_plot_rwr_attributes works", {
    expect_is(summary_plot_rwr_attributes(rwr_res_type), "ggplot")
    
    expect_is(summary_plot_rwr_attributes(rwr_res_type,color = list("1" = "red", "2"="blue", "3"="pink")),"ggplot")
    expect_is(summary_plot_rwr_attributes(rwr_res_type, seed.id = c("A","B")), "ggplot")
    expect_is(summary_plot_rwr_attributes(rwr_res_type, seed.type = c("1")), "ggplot")
    
    expect_is(summary_plot_rwr_attributes(rwr_res_type, seed.type = c("1"), plot = NA), "ggplot")
    expect_is(summary_plot_rwr_attributes(rwr_res_type, seed.type = c("1"), plot = FALSE), "ggplot")
    
    expect_is(summary_plot_rwr_attributes(rwr_res_type.list), "ggplot")
    
    expect_null(summary_plot_rwr_attributes(rwr_res_type, seed.type = c("f"), plot = FALSE))
    
})

test_that("plot_rwr_subnetwork fails on invalid input", {
    # X
    expect_error(plot_rwr_subnetwork(c()))
    
    # color
    expect_error(plot_rwr_subnetwork(X = rwr_res_type$A, color = 3), "'color' must be a named verctor or list", fixed = TRUE)
    expect_error(plot_rwr_subnetwork(X = rwr_res_type$A, color = data.frame()), "'color' must be a named verctor or list", fixed = TRUE)
    
    # plot, legend -> TRUE/FALSE 
    # ...
})

test_that("plot_rwr_subnetwork works", {
    # X
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A), "igraph")
    
    #color
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, color = list("1" = "red", "2"="blue", "3"="green")), "igraph")
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, color = list("1" = "red", "4"="blue", "3"="green")), "igraph")
    
    #plot
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, plot = FALSE), "igraph")
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, plot = NA), "igraph")
    
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, legend = FALSE), "igraph")
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, legend = NA), "igraph")
    expect_is(plot_rwr_subnetwork(X = rwr_res_type$A, legend = NULL), "igraph")
    
})