tests/testthat/test_em.R

In marcalva/diem: Debris-Containing Droplet Identification using EM

# Tests functions in em.R

context("EM")

test_that("EM flags errors when initialized", {
          expect_error(run_em(mb_small))
          mb_small <- set_debris_test_set(mb_small, verbose=FALSE)
          expect_error(run_em(mb_small))
          mb_small <- filter_genes(mb_small, verbose=FALSE)
          expect_error(run_em(mb_small))
})

mb_small <- set_debris_test_set(mb_small, min_counts = 5, verbose = FALSE)
mb_small <- filter_genes(mb_small, verbose = FALSE)
mb_small <- get_pcs(mb_small, min_genes=10, n_var_genes = 100)
mb_small <- init(mb_small, k_init = 10, verbose = FALSE)


test_that("EM works when initialized", {
          mb_small <- run_em(mb_small, verbose=FALSE, eps = 1e-4)
          mb_small <- assign_clusters(mb_small)
          mb_small <- estimate_dbr_score(mb_small, thresh_genes = 50)
          mb_small <- call_targets(mb_small, min_genes = 5, verbose=FALSE)
          emo <- mb_small@model
          expect_equal(ncol(emo$params$Alpha), length(emo$params$Pi))
          expect_true(emo$converged)
                                })