59ad1d186eb5a5bf3676de945f265b8b089508a1
[morpheus.git] / pkg / tests / testthat / test-alignMatrices.R
1 context("alignMatrices")
2
3 # Helper to generate a random series of matrices to align
4 .generateMatrices = function(d, K, N, noise)
5 {
6 matrices = list( matrix(runif(d*K, min=-1, max=1),ncol=K) ) #reference
7 for (i in 2:N)
8 {
9 matrices[[i]] <- matrices[[1]][,sample(1:K)]
10 if (noise)
11 matrices[[i]] = matrices[[i]] + matrix(rnorm(d*K, sd=0.05), ncol=K)
12 }
13 matrices
14 }
15
16 test_that("labelSwitchingAlign correctly aligns de-noised parameters",
17 {
18 N = 30 #number of matrices
19 d_K_list = list(c(2,2), c(5,3))
20 for (i in 1:2)
21 {
22 d = d_K_list[[i]][1]
23 K = d_K_list[[i]][2]
24
25 # 1] Generate matrix series
26 matrices_permut = .generateMatrices(d,K,N,noise=FALSE)
27
28 # 2] Call align function with mode=approx1
29 matrices_aligned =
30 alignMatrices(matrices_permut[2:N], ref=matrices_permut[[1]], ls_mode="approx1")
31
32 # 3] Check alignment
33 for (j in 2:N)
34 expect_equal(matrices_aligned[[j-1]], matrices_permut[[1]])
35
36 # 2bis] Call align function with mode=approx2
37 matrices_aligned =
38 alignMatrices(matrices_permut[2:N], ref=matrices_permut[[1]], ls_mode="approx2")
39
40 # 3bis] Check alignment
41 for (j in 2:N)
42 expect_equal(matrices_aligned[[j-1]], matrices_permut[[1]])
43 }
44 })
45
46 test_that("labelSwitchingAlign correctly aligns noisy parameters",
47 {
48 N = 30 #number of matrices
49 d_K_list = list(c(2,2), c(5,3))
50 for (i in 1:2)
51 {
52 d = d_K_list[[i]][1]
53 K = d_K_list[[i]][2]
54 max_error = d * 0.2 #TODO: what value to choose ?
55
56 # 1] Generate matrix series
57 matrices_permut = .generateMatrices(d,K,N,noise=TRUE)
58
59 # 2] Call align function
60 matrices_aligned = alignMatrices(matrices_permut, ref="mean", ls_mode="exact")
61
62 # 3] Check alignment
63 for (j in 2:N)
64 {
65 expect_that( norm(matrices_aligned[[j]] - matrices_permut[[1]]),
66 is_less_than(max_error) )
67 }
68 }
69 })