context("RWR")

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)

test_that("random_walk_with_restart fails with invalid input", {
    # X
    expect_error(random_walk_restart(X=NULL))
    expect_error(random_walk_restart(X=data.frame()))
    expect_error(random_walk_restart(X=list(3)))
    
    # seed
    expect_error(random_walk_restart(X=graph1, seed = 3))
    
    # r
    expect_error(random_walk_restart(X=graph1, r = -2), "'r' must be a numeric value between 0 and 1", fixed = TRUE)
    expect_error(random_walk_restart(X=graph1, r = matrix(2)), "'r' must be a numeric value", fixed = TRUE)
    
})

test_that("random_walk_with_restart works", {
    
    # X
    expect_is(random_walk_restart(graph1.list), "list.rwr")
    expect_is(random_walk_restart(graph1), "rwr")
    
    # seed
    expect_is(random_walk_restart(graph1, seed = 'A'), "rwr")
    expect_is(random_walk_restart(graph1.list, seed = "A"), "list.rwr")
    expect_is(random_walk_restart(graph1, seed = NULL), "rwr")

    # r
    expect_is(random_walk_restart(graph1, seed = NULL, r = 0.4), "rwr")
    
})



test_that("rwr_find_seeds_between_attributes fails with invalid input", {
    # X
    expect_error(rwr_find_seeds_between_attributes(X=NULL))
    expect_error(rwr_find_seeds_between_attributes(X=matrix()))
    
    # k
    expect_error(rwr_find_seeds_between_attributes(X=rwr_res, k = -5), "'k' must be a numeric value between 0 and 200", fixed = TRUE)
    expect_error(rwr_find_seeds_between_attributes(X=rwr_res, k = list(3)),"'k' must be a numeric value", fixed = TRUE)
    
    # seed / attribute
    expect_error(rwr_find_seeds_between_attributes(X=rwr_res, seed = matrix(3)), "'seed' must be a charactor vector", fixed = TRUE)
    expect_error(rwr_find_seeds_between_attributes(X=rwr_res, attribute = matrix(3)), "'attribute' must be a charactor vector", fixed = TRUE)
    expect_error(rwr_find_seeds_between_attributes(X=rwr_res, attribute = c("4", "5")), "invalid length", fixed = TRUE)
})


test_that("rwr_find_seeds_between_attributes works", {
    # X
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res), "rwr.attributes")
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res.list), "list.rwr.attributes")
    
    # seed
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res, seed = logical(0)), "rwr.attributes")
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res, attribute = logical(0)), "rwr.attributes")
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res, seed = "Z"), "rwr.attributes")
    
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res.list, seed = "Z"), "list.rwr.attributes")
    
    
    # k 
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res, k = 1) , "rwr.attributes")
    expect_is(rwr_find_seeds_between_attributes(X=rwr_res, k = NULL), "rwr.attributes")
    
})


test_that("rwr_find_closest_type fails with invalid input", {
    # X
    expect_error(rwr_find_closest_type(X=NULL))
    expect_error(rwr_find_closest_type(X=matrix()))
    
    # seed / attribute
    expect_error(rwr_find_closest_type(X=rwr_res, seed = matrix(3)), "'seed' must be a charactor vector", fixed = TRUE)
    expect_error(rwr_find_closest_type(X=rwr_res, attribute = matrix(3)), "'attribute' must be a charactor vector", fixed = TRUE)
    expect_error(rwr_find_closest_type(X=rwr_res, attribute = c("4", "5")), "invalid length", fixed = TRUE)
    
    # value
    expect_error(rwr_find_closest_type(X=rwr_res, attribute = NULL, seed = NULL, value = c("a", "b")), "invalid length", fixed = TRUE)
    expect_error(rwr_find_closest_type(X=rwr_res, attribute = NULL, seed = NULL, value = matrix(1,2)), "'value' must be a charactor vector", fixed = TRUE)
    
})


test_that("rwr_find_closest_type works", {
    # X
    expect_is(rwr_find_closest_type(X=rwr_res), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res.list), "list.rwr.closest")
    
    # seeds / attributes 
    expect_is(rwr_find_closest_type(X=rwr_res, seed = logical(0)), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res, attribute = logical(0)), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res, attribute = NULL, seed = NULL), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res, attribute = "NULL", seed = NULL), "rwr.closest")
    
    #value 
    expect_is(rwr_find_closest_type(X=rwr_res, attribute = "NULL", seed = NULL, value = NULL), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res, attribute = NULL, seed = NULL, value = "test"), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res, value = "test"), "rwr.closest")
    
    expect_is(rwr_find_closest_type(X=rwr_res, seed = "Z"), "rwr.closest")
    expect_is(rwr_find_closest_type(X=rwr_res.list, seed = "Z"), "list.rwr.closest")
    
})

# test_that("top_k_graph returns NULL", {
#     rwr_top_k_graph
#     rwr_top_k_graph(X = rwr_res$graph, RWRM_Result_Object = rwr_res$rwr, Seed = "A", k = 1)
# })