X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=pkg%2Ftests%2Ftestthat%2Ftest.computeFilaments.R;fp=pkg%2Ftests%2Ftestthat%2Ftest.computeFilaments.R;h=9de6274a3162840a9b8ffa444b8c47c07dcb2de8;hb=1e20780ee1505fac6c7ed68d340892c497524561;hp=0000000000000000000000000000000000000000;hpb=fa8078f9658dccadc0f9bcc7b54c13ecda621816;p=talweg.git

diff --git a/pkg/tests/testthat/test.computeFilaments.R b/pkg/tests/testthat/test.computeFilaments.R
new file mode 100644
index 0000000..9de6274
--- /dev/null
+++ b/pkg/tests/testthat/test.computeFilaments.R
@@ -0,0 +1,36 @@
+#TODO: toy dataset, check that indices returned are correct + colors
+
+context("Check that getParamsDirs behaves as expected")
+
+test_that("on input of sufficient size, beta is estimated accurately enough", {
+	n = 100000
+	d = 2
+	K = 2
+	Pr = c(0.5, 0.5)
+
+	betas_ref = array( c(1,0,0,1 , 1,-2,3,1), dim=c(2,2,2) )
+	for (i in 1:(dim(betas_ref)[3]))
+	{
+		beta_ref = betas_ref[,,i]
+		#all parameters are supposed to be of norm 1: thus, normalize beta_ref
+		norm2 = sqrt(colSums(beta_ref^2))
+		beta_ref = beta_ref / norm2[col(beta_ref)]
+
+		io = generateSampleIO(n, d, K, Pr, beta_ref)
+		beta = getParamsDirs(io$X, io$Y, K)
+		betas = .labelSwitchingAlign(
+			array( c(beta_ref,beta), dim=c(d,K,2) ), compare_to="first", ls_mode="exact")
+
+		#Some traces: 0 is not well estimated, but others are OK
+		cat("\n\nReference parameter matrix:\n")
+		print(beta_ref)
+		cat("Estimated parameter matrix:\n")
+		print(betas[,,2])
+		cat("Difference norm (Matrix norm ||.||_1, max. abs. sum on a column)\n")
+		diff_norm = norm(beta_ref - betas[,,2])
+		cat(diff_norm,"\n")
+
+		#NOTE: 0.5 is loose threshold, but values around 0.3 are expected...
+		expect_that( diff_norm, is_less_than(0.5) )
+	}
+})