write test.EMGrank.c
[valse.git] / src / test / test.selectiontotale.c
1 #include "selectiontotale.h"
2 #include "ioutils.h"
3
4 int main(int argc, char** argv)
5 {
6 // read dimensions
7 const int nbDims = 5;
8 int* dimensions = readArray_int("dimensions",&nbDims,1);
9 int n = dimensions[0];
10 int p = dimensions[1];
11 int m = dimensions[2];
12 int k = dimensions[3];
13 int L = dimensions[4];
14 free(dimensions);
15 int lengthOne = 1;
16
17 ////////////
18 // INPUTS //
19 ////////////
20
21 // phiInit
22 const int dimPhiInit[] = {p, m, k};
23 float* phiInit = readArray_real("phiInit",dimPhiInit,3);
24
25 // rhoInit
26 const int dimRhoInit[] = {m, m, k};
27 float* rhoInit = readArray_real("rhoInit",dimRhoInit,3);
28
29 // piInit
30 float* piInit = readArray_real("piInit",&k,1);
31
32 // gamInit
33 const int dimGamInit[] = {n, k};
34 float* gamInit = readArray_real("gamInit",dimGamInit, 2);
35
36 // min number of iterations
37 int* pmini = readArray_int("mini",&lengthOne,1);
38 int mini = *pmini;
39 free(pmini);
40
41 // max number of iterations
42 int* pmaxi = readArray_int("maxi",&lengthOne,1);
43 int maxi = *pmaxi;
44 free(pmaxi);
45
46 // gamma
47 float* pgamma = readArray_real("gamma",&lengthOne,1);
48 float gamma = *pgamma;
49 free(pgamma);
50
51 // lambda
52 float* glambda = readArray_real("glambda",&L,1);
53
54 // X
55 const int dimX[] = {n, p};
56 float* X = readArray_real("X",dimX,2);
57
58 // Y
59 const int dimY[] = {n, m};
60 float* Y = readArray_real("Y",dimY,2);
61
62 // seuil
63 float* pseuil = readArray_real("seuil",&lengthOne,1);
64 float seuil = *pseuil;
65 free(pseuil);
66
67 // tau
68 float* ptau = readArray_real("tau",&lengthOne,1);
69 float tau = *ptau;
70 free(ptau);
71
72 /////////////
73 // OUTPUTS //
74 /////////////
75
76 // A1
77 const int dimA[] = {p, m+1, L};
78 int* A1 = (int*)malloc(dimA[0]*dimA[1]*dimA[2]*sizeof(int));
79
80 // A2
81 int* A2 = (int*)malloc(dimA[0]*dimA[1]*dimA[2]*sizeof(int));
82
83 // rho
84 const int dimRho[] = {m, m, k, L};
85 float* Rho = (float*)malloc(dimRho[0]*dimRho[1]*dimRho[2]*dimRho[3]*sizeof(float));
86
87 // pi
88 const int dimPi[] = {k, L};
89 float* Pi = (float*)malloc(dimPi[0]*dimPi[1]*sizeof(float));
90
91 //////////////////////////////////////////////
92 // Main call to constructionModelesLassoMLE //
93 //////////////////////////////////////////////
94
95 selectiontotale(
96 phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,glambda,X,Y,seuil,tau,
97 A1,A2,Rho,Pi,
98 n,p,m,k,L);
99
100 // free input pointers
101 free(phiInit);
102 free(rhoInit);
103 free(piInit);
104 free(gamInit);
105 free(glambda);
106 free(X);
107 free(Y);
108
109 // Compare to reference outputs
110 int* ref_A1 = readArray_int("A1",dimA, 3);
111 compareArray_int("A1", A1, ref_A1, dimA[0]*dimA[1]*dimA[2]);
112 free(A1);
113 free(ref_A1);
114
115 // A2
116 int* ref_A2 = readArray_int("A2",dimA, 3);
117 compareArray_int("A2", A2, ref_A2, dimA[0]*dimA[1]*dimA[2]);
118 free(A2);
119 free(ref_A2);
120
121 // Rho
122 float* ref_Rho = readArray_real("Rho",dimRho, 4);
123 compareArray_real("Rho", Rho, ref_Rho, dimRho[0]*dimRho[1]*dimRho[2]*dimRho[3]);
124 free(Rho);
125 free(ref_Rho);
126
127 // Pi
128 float* ref_Pi = readArray_real("Pi",dimPi, 2);
129 compareArray_real("Pi", Pi, ref_Pi, dimPi[0]*dimPi[1]);
130 free(Pi);
131 free(ref_Pi);
132
133 return 0;
134 }