tests/testthat/test-specify_priors.R

In JANE: Just Another Latent Space Network Clustering Algorithm

## Testing specify_priors

test_that("specify_priors works", {
  
   # Simulate network
   mus <- matrix(c(-1,-1,1,-1,1,1), 
                 nrow = 3,
                 ncol = 2, 
                 byrow = TRUE)
   omegas <- array(c(diag(rep(7,2)),
                     diag(rep(7,2)), 
                     diag(rep(7,2))), 
                     dim = c(2,2,3))
   p <- rep(1/3, 3)
   beta0 <- 1.0
   sim_data <- JANE::sim_A(N = 100L, 
                           model = "RS",
                           mus = mus, 
                           omegas = omegas, 
                           p = p, 
                           params_LR = list(beta0 = beta0),
                           remove_isolates = TRUE)
                           
                           
   # Specify prior hyperparameters
   D <- 3L
   K <- 5L
   n_interior_knots <- 5L
   
   a <- rep(1, D)
   b <- 3
   c <- 4
   G <- 10*diag(D)
   nu <- rep(2, K)
   e <- rep(0.5, 1 + (n_interior_knots + 1))
   f <- diag(c(0.1, rep(0.5, n_interior_knots + 1)))
   
   expect_no_error({
     my_prior_hyperparameters <- specify_priors(D = D,
                                                K = K,
                                                model = "RS",
                                                n_interior_knots = n_interior_knots,
                                                a = a,
                                                b = b,
                                                c = c,
                                                G = G,
                                                nu = nu,
                                                e = e,
                                                f = f)
     
     # Run JANE on simulated data using supplied prior hyperparameters
     res <- JANE::JANE(A = sim_data$A,
                       D = D,
                       K = K,
                       initialization = "GNN",
                       model = "RS",
                       case_control = FALSE,
                       DA_type = "none",
                       control = list(priors = my_prior_hyperparameters))
   })

  
})