X-Git-Url: https://git.auder.net/?p=morpheus.git;a=blobdiff_plain;f=pkg%2Ftests%2Ftestthat%2Ftest-jointDiag.R;h=bb12a6c8a0b313245c24529bc8020160232bdd02;hp=0eb6ffd21968443f00472c535c47374fd665851a;hb=6dd5c2acccd10635449230faa824b7e8906911bf;hpb=bbdcfe44da4011574dabb19b4f83e2ab199c667a

diff --git a/pkg/tests/testthat/test-jointDiag.R b/pkg/tests/testthat/test-jointDiag.R
index 0eb6ffd..bb12a6c 100644
--- a/pkg/tests/testthat/test-jointDiag.R
+++ b/pkg/tests/testthat/test-jointDiag.R
@@ -3,48 +3,48 @@ context("jointDiag::ajd")
 #auxiliary to test diagonality
 .computeMuCheckDiag = function(X, Y, K, jd_method, β_ref)
 {
-	d = ncol(X)
-	#TODO: redundant code, same as computeMu() main method. Comments are stripped
-	M3 = .Moments_M3(X,Y)
-	M2_t = array(dim=c(d,d,d))
-	for (i in 1:d)
-	{
-		ρ = c(rep(0,i-1),1,rep(0,d-i))
-		M2_t[,,i] = .T_I_I_w(M3,ρ)
-	}
-	jd = jointDiag::ajd(M2_t, method=jd_method)
-	V = if (jd_method=="uwedge") jd$B else solve(jd$A)
-	M2_t = array(dim=c(d,d,K))
-	for (i in 1:K)
-		M2_t[,,i] = .T_I_I_w(M3,V[,i])
-	#END of computeMu() code
+  d = ncol(X)
+  #TODO: redundant code, same as computeMu() main method. Comments are stripped
+  M3 = .Moments_M3(X,Y)
+  M2_t = array(dim=c(d,d,d))
+  for (i in 1:d)
+  {
+    ρ = c(rep(0,i-1),1,rep(0,d-i))
+    M2_t[,,i] = .T_I_I_w(M3,ρ)
+  }
+  jd = jointDiag::ajd(M2_t, method=jd_method)
+  V = if (jd_method=="uwedge") jd$B else solve(jd$A)
+  M2_t = array(dim=c(d,d,K))
+  for (i in 1:K)
+    M2_t[,,i] = .T_I_I_w(M3,V[,i])
+  #END of computeMu() code
 
-	max_error = 0.5 #TODO: tune ?
-	invβ = MASS::ginv(β_ref)
-	for (i in 1:K)
-	{
-		shouldBeDiag = invβ %*% M2_t[,,i] %*% t(invβ)
-		expect_that(
-			mean( abs(shouldBeDiag[upper.tri(shouldBeDiag) | lower.tri(shouldBeDiag)]) ),
-			is_less_than(max_error) )
-	}
+  max_error = 0.5 #TODO: tune ?
+  invβ = MASS::ginv(β_ref)
+  for (i in 1:K)
+  {
+    shouldBeDiag = invβ %*% M2_t[,,i] %*% t(invβ)
+    expect_that(
+      mean( abs(shouldBeDiag[upper.tri(shouldBeDiag) | lower.tri(shouldBeDiag)]) ),
+      is_less_than(max_error) )
+  }
 }
 
 test_that("'jedi' and 'uwedge' joint-diagonalization methods return a correct matrix",
 {
-	n = 10000
-	d_K = list( c(2,2), c(5,3), c(20,13) )
+  n = 10000
+  d_K = list( c(2,2), c(5,3), c(20,13) )
 
-	for (dk_index in 1:length(d_K))
-	{
-		d = d_K[[dk_index]][1]
-		K = d_K[[dk_index]][2]
-		#NOTE: sometimes large errors if pr is not balanced enough (e.g. random);
-		#      same note for β. However we could be more random than that...
-		β_ref = rbind(diag(K),matrix(0,nrow=d-K,ncol=K))
-		io = generateSampleIO(n, p=rep(1/K,K-1), β=β_ref, rep(0,K), link="logit")
-		.computeMuCheckDiag(io$X, io$Y, K, jd_method="uwedge", β_ref)
-		#TODO: some issues with jedi method (singular system)
-		#.computeMuCheckDiag(io$X, io$Y, K, jd_method="jedi", β_ref)
-	}
+  for (dk_index in 1:length(d_K))
+  {
+    d = d_K[[dk_index]][1]
+    K = d_K[[dk_index]][2]
+    #NOTE: sometimes large errors if pr is not balanced enough (e.g. random);
+    #      same note for β. However we could be more random than that...
+    β_ref = rbind(diag(K),matrix(0,nrow=d-K,ncol=K))
+    io = generateSampleIO(n, p=rep(1/K,K-1), β=β_ref, rep(0,K), link="logit")
+    .computeMuCheckDiag(io$X, io$Y, K, jd_method="uwedge", β_ref)
+    #TODO: some issues with jedi method (singular system)
+    #.computeMuCheckDiag(io$X, io$Y, K, jd_method="jedi", β_ref)
+  }
 })