# Updating
# 1. compare leidenbase output using the code in leidenbase/tests/cpp_tests
#      o  check the NOTES file in leidenbase/tests/cpp_tests
#      o  compare leidenbase C++ to python leidenalg
#      o  compare leidenbase R to python leidenalg
# 2. rebuild the expected values for devtools::test() by setting run_setup
#    below to TRUE and running
#      cd leidenbase
#      R
#      devtools::test()
#   This should update the files in leidenbase/inst/testdata.
#   Remember to reset run_setup to FALSE.


run_setup <- FALSE
dst_dir <- paste0(system.file(package='leidenbase'), '/testdata')
evf <- paste0(dst_dir, '/expected_values.txt')

#
# Get igraph object
#
fname <- 'igraph_n1500_edgelist.txt.gz'
fpath <- system.file( 'testdata', fname, package = 'leidenbase' )
if( length( fpath ) > 0 )
{
  zfp <- gzfile( fpath )
  igraph <- igraph::read_graph( file = zfp, format='edgelist', n=1500 )
} else {
  stop( 'Error: unable to find file ', fname, '\n' )
}

if(run_setup) {
  system(paste0('rm ',  evf))
} else {
  source(evf)
}

#
# Test partition types.
#
testthat::test_that( 'CPMVertexPartition membership and quality',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'CPMVertexPartition', seed = 123456, resolution_parameter = 0.5 )
  t01_f01 <- 'test01_file01.txt'
  t01_v01 <- res$membership
  t01_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t01_f01)
  if(run_setup) {
    write.table(t01_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t01_v02_expect <- %.2f\n", t01_v02), file=evf, append=TRUE)
  }
  else {
    t01_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t01_v01, t01_v01_expect)))
    expect_equal(t01_v02, t01_v02_expect, tolerance = 1.0e-2)
  }
})

testthat::test_that( 'ModularityVertexPartition membership and quality',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'ModularityVertexPartition', seed = 123456, resolution_parameter = 0.5 )
  t02_f01 <- 'test02_file01.txt'
  t02_v01 <- res$membership
  t02_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t02_f01)
  if(run_setup) {
    write.table(t02_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t02_v02_expect <- %.4f\n", t02_v02), file=evf, append=TRUE)
  }
  else {
    t02_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t02_v01, t02_v01_expect)))
    expect_equal(t02_v02, t02_v02_expect, tolerance = 1.0e-2)
  }
})

testthat::test_that( 'RBERVertexPartition membership and quality',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBERVertexPartition', seed = 123456, resolution_parameter = 0.5 )
  t03_f01 <- 'test03_file01.txt'
  t03_v01 <- res$membership
  t03_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t03_f01)
  if(run_setup) {
    write.table(t03_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t03_v02_expect <- %.2f\n", t03_v02), file=evf, append=TRUE)
  }
  else {
    t03_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t03_v01, t03_v01_expect)))
    expect_equal(t03_v02, t03_v02_expect, tolerance = 1.0e-0)
  }
})

testthat::test_that( 'SignificanceVertexPartition membership and quality', 
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'SignificanceVertexPartition', seed = 123456, resolution_parameter = 0.5 )
  t04_f01 <- 'test04_file01.txt'
  t04_v01 <- res$membership
  t04_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t04_f01)
  if(run_setup) {
    write.table(t04_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t04_v02_expect <- %.2f\n", t04_v02), file=evf, append=TRUE)
  }
  else {
    t04_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t04_v01, t04_v01_expect)))
    expect_equal(t04_v02, t04_v02_expect, tolerance = 1.0e-2)
  }
})

testthat::test_that( 'SurpriseVertexPartition membership and quality', 
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'SurpriseVertexPartition', seed = 123456, resolution_parameter = 0.5 )
  t05_f01 <- 'test05_file01.txt'
  t05_v01 <- res$membership
  t05_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t05_f01)
  if(run_setup) {
    write.table(t05_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t05_v02_expect <- %.2f\n", t05_v02), file=evf, append=TRUE)
  }
  else {
    t05_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t05_v01, t05_v01_expect)))
    expect_equal(t05_v02, t05_v02_expect, tolerance = 1.0e-0)
  }
})


#
# Test resolution_parameter and num_iter.
#
testthat::test_that( 'resolution_parameter parameter',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBConfigurationVertexPartition', resolution_parameter = 1, seed = 123456 )
  t06_f01 <- 'test06_file01.txt'
  t06_v01 <- res$membership
  t06_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t06_f01)
  if(run_setup) {
    write.table(t06_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t06_v02_expect <- %.2f\n", t06_v02), file=evf, append=TRUE)
  }
  else {
    t06_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t06_v01, t06_v01_expect)))
    expect_equal(t06_v02, t06_v02_expect, tolerance = 1.0e-0)
  }
})

testthat::test_that( 'num_iter parameter',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBConfigurationVertexPartition', num_iter = 5, seed = 123456, resolution_parameter = 0.5 )
  t07_f01 <- 'test07_file01.txt'
  t07_v01 <- res$membership
  t07_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t07_f01)
  if(run_setup) {
    write.table(t07_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t07_v02_expect <- %.2f\n", t07_v02), file=evf, append=TRUE)
  }
  else {
    t07_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t07_v01, t07_v01_expect)))
    expect_equal(t07_v02, t07_v02_expect, tolerance = 1.0e-0)
  }
})


#
# Test initial_membership
#
testthat::test_that( 'initial_membership parameter',
{
  inimem <- as.integer( seq( 1, 1500 ) )
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBConfigurationVertexPartition', initial_membership = inimem, seed = 123456, resolution_parameter = 0.5 )
  t08_f01 <- 'test08_file01.txt'
  t08_v01 <- res$membership
  t08_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t08_f01)
  if(run_setup) {
    write.table(t08_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t08_v02_expect <- %.2f\n", t08_v02), file=evf, append=TRUE)
  }
  else {
    t08_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t08_v01, t08_v01_expect)))
    expect_equal(t08_v02, t08_v02_expect, tolerance = 1.0e-0)
  }
})


#
# Test edge_weights
#
testthat::test_that( 'edge_weights parameter',
{
  edgwgt <- as.double( rep( 1.0, 61636 ) )
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBConfigurationVertexPartition', edge_weights = edgwgt, seed = 123456, resolution_parameter = 0.5 )
  t09_f01 <- 'test09_file01.txt'
  t09_v01 <- res$membership
  t09_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t09_f01)
  if(run_setup) {
    write.table(t09_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t09_v02_expect <- %.2f\n", t09_v02), file=evf, append=TRUE)
  }
  else {
    t09_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t09_v01, t09_v01_expect)))
    expect_equal(t09_v02, t09_v02_expect, tolerance = 1.0e-0)
  }
})


#
# Test node_sizes
#
testthat::test_that( 'node_sizes parameter',
{
  nodsiz <- as.integer( rep( 1, 1500 ) )
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'RBConfigurationVertexPartition', node_sizes = nodsiz, seed = 123456, resolution_parameter = 0.5 )
  t10_f01 <- 'test10_file01.txt'
  t10_v01 <- res$membership
  t10_v02 <- res$quality
  fpath <- paste0(dst_dir, '/', t10_f01)
  if(run_setup) {
    write.table(t10_v01, file=fpath, row.names=FALSE, col.names=FALSE)
    cat(sprintf("t10_v02_expect <- %.2f\n", t10_v02), file=evf, append=TRUE)
  }
  else {
    t10_v01_expect <- read.table(fpath)[['V1']]
    expect_true(isTRUE(all.equal(t10_v01, t10_v01_expect)))
    expect_equal(t10_v02, t10_v02_expect, tolerance = 1.0e-0)
  }
})

#
# Test modularity and significance return values.
#
testthat::test_that( 'modularity and significance return values',
{
  res <- leidenbase::leiden_find_partition( igraph, partition_type = 'CPMVertexPartition', seed = 123456, resolution_parameter = 0.1 )
  t11_v01 <- res$modularity
  t11_v02 <- res$significance
  if(run_setup) {
    cat(sprintf("t11_v01_expect <- %.4f\n", t11_v01), file=evf, append=TRUE)
    cat(sprintf("t11_v02_expect <- %.2f\n", t11_v02), file=evf, append=TRUE)
  }
  else {
    expect_equal(t11_v01, t11_v01_expect, tolerance = 1.0e-2)
    expect_equal(t11_v02, t11_v02_expect, tolerance = 1.0e-2)
  }
})

#
# Test parameter errors
#
egraph <- ''
testthat::test_that( 'igraph parameter error',
{
  expect_error( leidenbase::leiden_find_partition( egraph ) )
})

inimem <- inimem <- as.integer( seq( 1, 1400 ) )
testthat::test_that( 'initial_membership parameter error',
{
  expect_error( leidenbase::leiden_find_partition( igraph, initial_membership = inimem, resolution_parameter = 0.5 ) )
})

edgwgt <- as.double( rep( 1.0, 61630 ) )
testthat::test_that( 'edge_weights parameter error',
{
  expect_error( leidenbase::leiden_find_partition( igraph, edge_weights = edgwgt, resolution_parameter = 0.5 ) )
})

nodsiz <- inimem <- as.integer( seq( 1, 1400 ) )
testthat::test_that( 'node_sizes parameter error',
{
  expect_error( leidenbase::leiden_find_partition( igraph, node_sizes = nodsiz, resolution_parameter = 0.5 ) )
})

testthat::test_that( 'resolution_parameter parameter error',
{
  expect_error( leidenbase::leiden_find_partition( igraph, resolution_parameter = 0.0 ) )
})

testthat::test_that( 'num_iter parameter error',
{
  expect_error( leidenbase::leiden_find_partition( igraph, num_iter = 0, resolution_parameter = 0.5 ) )
})