X-Git-Url: https://git.auder.net/?p=valse.git;a=blobdiff_plain;f=test%2Fhelper.R;h=21d00f2b6698671c2c5d3af50dd4273b733211b3;hp=8ec122bd368f49e5127db6ffc2b5ac9bb8582353;hb=bb64f5cbaa3f0d4cc49a0325ca24d4f41f90ccc3;hpb=3921ba9b5ea85bcc190245ac7da9ee9da1658b9f

diff --git a/test/helper.R b/test/helper.R
index 8ec122b..21d00f2 100644
--- a/test/helper.R
+++ b/test/helper.R
@@ -11,19 +11,20 @@ generateXYdefault = function(n, p, m, k)
 {
 	meanX = rep(0, p)
 	covX = diag(p)
-	covY = array(dim=c(m,m,k))
-	for(r in 1:k)
-		covY[,,r] = diag(m)
-	π = rep(1./k,k)
+	covY = diag(m)
+	ω = rep(1./k,k)
 	#initialize beta to a random number of non-zero random value
 	β = array(0, dim=c(p,m,k))
 	for (j in 1:p)
 	{
 		nonZeroCount = sample(1:m, 1)
-		β[j,1:nonZeroCount,] = matrix(runif(nonZeroCount*k), ncol=k)
+		if (nonZeroCount >= 2)
+			β[j,1:nonZeroCount,] = matrix(runif(nonZeroCount*k), ncol=k)
+		else
+			β[j,1,] = runif(k)
 	}
 
-	sample_IO = generateXY(n, π, meanX, β, covX, covY)
+	sample_IO = generateXY(n, ω, meanX, β, covX, covY)
 	return (list(X=sample_IO$X,Y=sample_IO$Y))
 }