#library(lolog)
library(testthat)
library(ergm)
library(network)

test_that("Stats", {
  
  data(ukFaculty)
  data(florentine)
  data(lazega)
  
  Group <- ukFaculty %v% "Group"
  Group[is.na(Group)] <- 4
  ukFaculty %v% "Group" <- Group
  ukFaculty %v% "GroupC" <- as.character(Group)
  
  es <- summary(ukFaculty ~ nodefactor("GroupC",4))
  lss <- calculateStatistics(ukFaculty ~ nodeFactor("GroupC"))
  expect_true(all(es == lss))
  
  
  es <- summary(ukFaculty ~ mutual)
  lss <- calculateStatistics(ukFaculty ~ mutual)
  expect_true(all(es == lss))
  
  
  wabsdiff<-outer(flomarriage %v% "wealth", flomarriage %v% "wealth",function(x,y){abs(x-y)})
  es <- summary(flomarriage ~ edgecov(wabsdiff))
  lss <- calculateStatistics(flomarriage ~ edgeCov(wabsdiff))
  expect_true(all(es == lss))
  
  es <- summary(flomarriage ~ absdiff("wealth"))
  lss <- calculateStatistics(flomarriage ~ absDiff("wealth"))
  expect_true(all(es == lss))
  
  es <- summary(flomarriage ~ absdiff("wealth",3))
  lss <- calculateStatistics(flomarriage ~ absDiff("wealth",3))
  expect_true(all(es == lss))
  
  lss <- calculateStatistics(lazega ~ degree(0:10) + esp(0:10, 1))
  expect_equivalent(lss, c(2, 3, 2, 4, 2, 4, 4, 1, 1, 5, 1, 5, 16, 29, 17, 23, 
                           11, 10, 4, 0, 0, 0))
  
  lss <- calculateStatistics(ukFaculty ~ degree(direction="in", d=0:10) + degree(0:10, direction="out"))
  expect_equivalent(lss, c(0, 1, 3, 0, 5, 4, 10, 7, 7, 11, 5, 1, 3, 5, 7, 6, 
                           4, 9, 4, 4, 1, 5))
  
  es <- summary(ukFaculty ~ twopath())
  lss <- calculateStatistics(ukFaculty ~ twoPath())
  expect_true(all(es == lss))
  
  es <- summary(lazega ~ twopath())
  lss <- calculateStatistics(lazega ~ twoPath())
  expect_true(all(es == lss))
  
  i <- c(1)
  j <- c(1)
  x <- c(1)
  A <- Matrix::sparseMatrix(i, j, x = x,dims=c(36,36))
  A[1,1] <- 0
  for(i in 1:20){
    i <- sample(36,1)
    j <- sample(36,1)
    A[j,i] <- A[i,j] <- runif(1)
  }
  
  f1 <- calculateStatistics(lazega ~ edges() + edgeCov(as.matrix(A)))
  f2 <- calculateStatistics(lazega ~ edges() + edgeCovSparse(A))
  expect_equivalent(f1,f2)
  
  data(sampson)
  f <- calculateStatistics(samplike ~ nodeMix("group"))
  expect_equivalent(f, c(nodemix.Loyal.Loyal = 23, nodemix.Loyal.Outcasts = 2, nodemix.Loyal.Turks = 9, 
                         nodemix.Outcasts.Loyal = 1, nodemix.Outcasts.Outcasts = 10, nodemix.Outcasts.Turks = 7, 
                         nodemix.Turks.Loyal = 5, nodemix.Turks.Outcasts = 1, nodemix.Turks.Turks = 30
  ))
})