tests/testthat/test_dataframe_to_igraph.R

In tbonne/netTS: netTS: network time series

context("input to igraph tests")
library(netTS)
library(lubridate)
library(igraph)
library(dplyr)

#make up a simple dataframe of observations
test.df <- data.frame(from = c("amy", "greg", "sarah", "olga"),
                      to = c("greg","amy", "amy", "greg"),
                      weight= c(1,2,3,4),
                      date = c("2014-10-23","2014-10-23","2014-10-23","2014-11-23" )
                      )

#create graph
out.graph.undirected <- create.a.network(test.df, directed=FALSE)
out.graph.directed <- create.a.network(test.df, directed=TRUE)

out.graph.SRI.un <- create.a.network.SRI(test.df, directed=FALSE)
out.graph.SRI.dir <- create.a.network.SRI(test.df, directed=TRUE)

test.ordered <- order_events(test.df)


#test that it worked properly
test_that("Out network is equal to expectations from custom dataframe", {
  expect_equal(vcount(out.graph.undirected), 4)
  expect_equal(vcount(out.graph.directed), 4)
  expect_equal(ecount(out.graph.undirected), 3)
  expect_equal(ecount(out.graph.directed), 4)
  expect_equal(sum(E(out.graph.undirected)$weight - c(3,3,4)),0 )
  expect_equal(sum(E(out.graph.directed)$weight - c(1,2,4,3)) , 0)
  expect_equal(test.ordered[3,1] , "amy")
  expect_equal(test.ordered[4,2] , "olga")
  expect_equal(E(out.graph.SRI.dir)[get.edge.ids(out.graph.SRI.dir,vp=c("olga","greg"))]$weight, 4/5)
  expect_equal(E(out.graph.SRI.dir)[get.edge.ids(out.graph.SRI.dir,vp=c("greg","amy"))]$weight, 2/5)
  expect_equal(E(out.graph.SRI.un)[get.edge.ids(out.graph.SRI.un,vp=c("olga","greg"))]$weight, 4/7)
  expect_equal(E(out.graph.SRI.un)[get.edge.ids(out.graph.SRI.un,vp=c("greg","amy"))]$weight, 3/10)
})