Skip to content

Commit

Permalink
fixes tests
Browse files Browse the repository at this point in the history
  • Loading branch information
KrystynaGrzesiak committed Sep 14, 2024
1 parent add67a3 commit 19b0b3c
Showing 1 changed file with 25 additions and 22 deletions.
47 changes: 25 additions & 22 deletions tests/testthat/test-filtering.R
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ test_that("list filtering methods works", {
"filter_praznik", "filter_quipt"))
})


set.seed(12)

n_seq <- 50
sequence_length <- 10
Expand All @@ -24,55 +24,58 @@ test_that("quipt filter works", {
res <- filter_quipt(target, kmers)

expect_equal(res[1:10],
c(a_0 = 0.433460286708944, d_0 = 0.537216985854405,
b_0 = 0.862413139242882, d.c_0 = 1, d.b_0 = 0.904876795162511,
d.d_0 = 0.0668944277862377, b.a_0 = 0.904876795162511,
a.d_0 = 0.741734233008649, a.a_0 = 0.904876795162511,
c.b_0 = 0.904876795162511))
c(d_0 = 0.75732779982674, b_0 = 0.728316101771109,
d.c_0 = 1, d.a_0 = 1, a.b_0 = 0.728316101771109, d.d_0 = 1,
b.a_0 = 1, b.c_0 = 1, a.d_0 = 0.728316101771109, c.d_0 = 1))
})


test_that("chisq filter works", {
res <- filter_chisq(target, kmers)

expect_equal(res[1:10],
c(a_0 = 0.359163595525953, d_0 = 0.619081488098256,
b_0 = 0.772148036951171, c_0 = 1, d.c_0 = 0.999999999999999,
d.b_0 = 0.693010969821401, d.d_0 = 0.00137293131428169,
b.a_0 = 0.504838137702313, a.d_0 = 0.113732669593295,
a.a_0 = 0.438514651784921))
c(a_0 = 1, d_0 = 0.885992079772807, b_0 = 0.137259222349999,
c_0 = 1, d.c_0 = 0.710251149725874, d.a_0 = 0.470784349666462,
a.b_0 = 0.056129794661529, d.d_0 = 0.793891166336373,
b.a_0 = 0.514976937835016, b.c_0 = 1))
})


test_that("fcbf filter works", {
res <- filter_fcbf(target, kmers)
res <- filter_fcbf(target, kmers, thresh = 0.1)

expect_equal(res, c("d.d.c_0.4", "d.d.b_0.1"))
expect_equal(res, c("c.d.c_1.3", "a.d.d.c_0.2.1", "b.d.a_2.1", "d.a.d_4.1", "b.a_6",
"a.b.b_0.1", "d.a.b_0.1", "c.a.a_2.2", "d.c.b_1.4", "d.c.c_3.0",
"c.d.c_3.1", "b.c.c_0.0", "d.b.d_0.1", "c.a.d.c_1.0.3", "b.a.b_2.0",
"a.a.b_3.0", "a.c.a_2.1", "c.d.d_0.2", "a.b.d_0.3", "c.c.d_0.3",
"a.b.a_1.3", "d.c.d_2.3", "d.c.a_1.4", "d.d.d.d_0.1.1"))
})


test_that("FSelectorRcpp filter works", {
res <- filter_ig(target, kmers, method = "infogain")

expect_equal(res[1:5], c(a_0 = 0.0926487792836429, d_0 = NaN, b_0 = NaN,
c_0 = NaN, d.a_0 = 0.0318668917281004))
expect_equal(res[1:5], c(a_0 = NaN, d_0 = 0.0243373426620957,
b_0 = 0.0545796620922217, c_0 = NaN,
d.c_0 = 0.00758121297699884))
})


test_that("FSelectorRcpp filter works", {
res <- filter_ig(target, kmers, method = "infogain")

expect_equal(res[1:5], c(a_0 = 0.0926487792836429, d_0 = NaN, b_0 = NaN,
c_0 = NaN, d.a_0 = 0.0318668917281004))
expect_equal(res[1:5], c(a_0 = NaN, d_0 = 0.0243373426620957,
b_0 = 0.0545796620922217, c_0 = NaN,
d.c_0 = 0.00758121297699884))
})


test_that("praznik filter works", {
res <- filter_praznik(target, kmers, method = "MIM")

expect_equal(res[1:5], c(d.d.c_1.4 = 0.309448173049744,
b.c.a.d_0.0.1 = 0.309448173049744,
b.c.b.c_0.5.1 = 0.309448173049744,
b.c.d.b_0.2.2 = 0.309448173049744,
b.c.d_0.2 = 0.29557289101432))
expect_equal(res[1:5], c(c.d.c_1.3 = 0.127675835930895,
a.d.d.c_0.2.1 = 0.127675835930895,
d.d.a_2.2 = 0.111515430511026,
c.a.d.d_0.0.0 = 0.111515430511026,
c.c_5 = 0.101033251383654))
})

0 comments on commit 19b0b3c

Please sign in to comment.