tests/testthat/test-BIC.R

In gmgm: Gaussian Mixture Graphical Model Learning and Inference

gmm_1 <- list(alpha = c(0.4, 0.6),
              mu = matrix(c(2, 0, 3, 1), 2, dimnames = list(c("B", "A"), NULL)),
              sigma = list(matrix(c(2, 1, 1, 1), 2,
                                  dimnames = list(c("B", "A"), c("B", "A"))),
                           matrix(c(5, 3, 3, 2), 2,
                                  dimnames = list(c("B", "A"), c("B", "A")))))
class(gmm_1) <- "gmm"
gmm_2 <- list(alpha = c(0.4, 0.6),
              mu = matrix(c(0, 1), 1, dimnames = list("A", NULL)),
              sigma = list(matrix(1, dimnames = list("A", "A")),
                           matrix(2, dimnames = list("A", "A"))))
class(gmm_2) <- "gmm"
gmm_3 <- list(alpha = c(0.4, 0.6),
              mu = matrix(c(2, 0, 3, 1), 2,
                          dimnames = list(c("B", "A.1"), NULL)),
              sigma = list(matrix(c(2, 1, 1, 1), 2,
                                  dimnames = list(c("B", "A.1"),
                                                  c("B", "A.1"))),
                           matrix(c(5, 3, 3, 2), 2,
                                  dimnames = list(c("B", "A.1"),
                                                  c("B", "A.1")))))
class(gmm_3) <- "gmm"

gmbn_1 <- list(A = gmm_2, B = gmm_1)
class(gmbn_1) <- "gmbn"
gmbn_2 <- list(A = gmm_2, B = gmm_3)
class(gmbn_2) <- "gmbn"

gmdbn_1 <- list(b_1 = gmbn_1, b_2 = gmbn_1)
class(gmdbn_1) <- "gmdbn"
gmdbn_2 <- list(b_1 = gmbn_1, b_2 = gmbn_2)
class(gmdbn_2) <- "gmdbn"

test_that("compute the BIC of a gmm object", {
  expect_equal(BIC(gmm_1, data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6))),
               - 117.7619, tolerance = 0.01)
  expect_equal(BIC(gmm_1,
                   matrix(c(0, 3, 6, 9, 4, 7, 1, 6), 4,
                          dimnames = list(NULL, c("A", "B")))),
               - 117.7619, tolerance = 0.01)
})

test_that("compute the BIC of a gmm object with missing values", {
  expect_equal(BIC(gmm_1, data.frame(A = c(0, 3, NA, NA), B = c(4, 7, 1, NA))),
               as.numeric(NA))
})

test_that("compute the BIC of a gmm object with extra columns", {
  expect_equal(BIC(gmm_1,
                   data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6),
                              C = c(0, 0, 0, 0))),
               - 117.7619, tolerance = 0.01)
})

test_that("compute the BIC of a gmm object with no row", {
  expect_equal(BIC(gmm_1, data.frame(A = numeric(), B = numeric())), Inf)
  expect_equal(BIC(gmm_1, data.frame(A = logical(), B = logical())), Inf)
})

test_that("compute the non-regularized BIC of a gmm object", {
  expect_equal(BIC(gmm_1, data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6)),
                   regul = NULL),
               - 117.7119, tolerance = 0.01)
})

test_that("compute the conditional BIC of a gmm object", {
  expect_equal(BIC(gmm_1,
                   data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6)), y = "B"),
               - 87.92791, tolerance = 0.01)
})

test_that("compute the BIC of a gmbn object", {
  expect_equal(BIC(gmbn_1, data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6))),
               list(global = - 121.1777,
                    local = c(A = - 33.24975, B = - 87.92791)),
               tolerance = 0.01)
})

test_that("compute the BIC of a gmbn object with missing values", {
  expect_equal(BIC(gmbn_1, data.frame(A = c(0, 3, NA, NA), B = c(4, 7, 1, NA))),
               list(global = as.numeric(NA),
                    local = c(A = as.numeric(NA), B = as.numeric(NA))))
})

test_that("compute the BIC of a gmbn object with extra columns", {
  expect_equal(BIC(gmbn_1,
                   data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6),
                              C = c(0, 0, 0, 0))),
               list(global = - 121.1777,
                    local = c(A = - 33.24975, B = - 87.92791)),
               tolerance = 0.01)
})

test_that("compute the BIC of a gmbn object with no row", {
  expect_equal(BIC(gmbn_1, data.frame(A = numeric(), B = numeric())),
               list(global = Inf, local = c(A = Inf, B = Inf)))
  expect_equal(BIC(gmbn_1, data.frame(A = logical(), B = logical())),
               list(global = Inf, local = c(A = Inf, B = Inf)))
})

test_that("compute the BIC of a temporal gmbn object with one observation sequence", {
  expect_equal(BIC(gmbn_2, data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6))),
               list(global = - 78.59241,
                    local = c(A = - 31.29746, B = - 47.29494)),
               tolerance = 0.01)
  expect_equal(BIC(gmbn_2, data.frame(A = c(0, 3, 6, 9), B = c(4, 7, 1, 6)),
                   col_seq = character()),
               list(global = - 78.59241,
                    local = c(A = - 31.29746, B = - 47.29494)),
               tolerance = 0.01)
})

test_that("compute the BIC of a temporal gmbn object with several observation sequences", {
  expect_equal(BIC(gmbn_2,
                   data.frame(seq = c(1, 1, 2, 2), A = c(0, 3, 6, 9),
                              B = c(4, 7, 1, 6)),
                   col_seq = "seq"),
               list(global = - 54.81646,
                    local = c(A = - 22.25747, B = - 32.55898)),
               tolerance = 0.01)
})

test_that("compute the BIC of a temporal gmbn object with duplicated observation sequence column names", {
  expect_equal(BIC(gmbn_2,
                   data.frame(seq = c(1, 1, 2, 2), A = c(0, 3, 6, 9),
                              B = c(4, 7, 1, 6)),
                   col_seq = c("seq", "seq")),
               list(global = - 54.81646,
                    local = c(A = - 22.25747, B = - 32.55898)),
               tolerance = 0.01)
})

test_that("compute the BIC of a gmdbn object", {
  expect_equal(BIC(gmdbn_1,
                   data.frame(seq = c(1, 1, 2, 2), A = c(0, 3, 6, 9),
                              B = c(4, 7, 1, 6)),
                   col_seq = "seq"),
               list(global = - 121.1777,
                    local = list(b_1 = c(A = - 10.99228, B = - 44.54471),
                                 b_2 = c(A = - 22.25747, B = - 43.3832))),
               tolerance = 0.01)
})

test_that("compute the BIC of a gmdbn object", {
  expect_equal(BIC(gmdbn_2,
                   data.frame(seq = c(1, 1, 2, 2), A = c(0, 3, 6, 9),
                              B = c(4, 7, 1, 6)),
                   col_seq = "seq"),
               list(global = - 110.3534,
                    local = list(b_1 = c(A = - 10.99228, B = - 44.54471),
                                 b_2 = c(A = - 22.25747, B = - 32.55898))),
               tolerance = 0.01)
})