tests/testthat/test_init.R

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

# Tests functions in init_clust.R

context("Initialization")

#rf <- system.file("data", "mb_small.rda", package="diem")
#load(rf)

mb_small <- set_debris_test_set(mb_small, min_counts=100, verbose = FALSE)

expect_equal(sum(mb_small@droplet_data[,"total_counts"] >= 100), nrow(mb_small@test_data))
expect_equal(nrow(mb_small@test_data), length(mb_small@test_set))
expect_equal(nrow(mb_small@droplet_data)-nrow(mb_small@test_data), length(mb_small@bg_set))

mb_small <- filter_genes(mb_small, verbose = FALSE)

test_that("initialization works", {
         mb_small <- get_pcs(mb_small, n_var = 200, n_pcs = 30)
         expect_true( ncol(mb_small@pcs) == 30 ) 
         expect_equal(length(mb_small@vg), 200)

         mb_small <- init(mb_small, k_init = 15, verbose = FALSE)
         ic <- mb_small@model
         expect_equal( ncol(ic$params$Alpha), length(ic$params$Pi) )
})