cosmetics
[valse.git] / test / test.EMGLLF.c
1 #include "EMGLLF.h"
2 #include "test_utils.h"
3 #include <stdlib.h>
4
5 int main(int argc, char** argv)
6 {
7 int* dimensions = readArray_int("dimensions");
8 int n = dimensions[0];
9 int p = dimensions[1];
10 int m = dimensions[2];
11 int k = dimensions[3];
12 free(dimensions);
13
14 ////////////
15 // INPUTS //
16 Real* phiInit = readArray_real("phiInit");
17 Real* rhoInit = readArray_real("rhoInit");
18 Real* piInit = readArray_real("piInit");
19 Real* gamInit = readArray_real("gamInit");
20 int mini = read_int("mini");
21 int maxi = read_int("maxi");
22 Real gamma = read_real("gamma");
23 Real lambda = read_real("lambda");
24 Real* X = readArray_real("X");
25 Real* Y = readArray_real("Y");
26 Real tau = read_real("tau");
27 ////////////
28
29 /////////////
30 // OUTPUTS //
31 Real* phi = (Real*)malloc(p*m*k*sizeof(Real));
32 Real* rho = (Real*)malloc(m*m*k*sizeof(Real));
33 Real* pi = (Real*)malloc(k*sizeof(Real));
34 Real llh;
35 Real* S = (Real*)malloc(p*m*k*sizeof(Real));
36 int* affec = (int*)malloc(n*sizeof(int));
37 /////////////
38
39 ////////////////////
40 // Call to EMGLLF //
41 EMGLLF_core(phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,lambda,X,Y,tau,
42 phi,rho,pi,&llh,S,affec,
43 n,p,m,k);
44 ////////////////////
45
46 free(phiInit);
47 free(rhoInit);
48 free(piInit);
49 free(gamInit);
50 free(X);
51 free(Y);
52
53 // Compare to reference outputs
54 Real* ref_phi = readArray_real("phi");
55 compareArray_real("phi", phi, ref_phi, p*m*k);
56 free(phi);
57 free(ref_phi);
58
59 Real* ref_rho = readArray_real("rho");
60 compareArray_real("rho", rho, ref_rho, m*m*k);
61 free(rho);
62 free(ref_rho);
63
64 Real* ref_pi = readArray_real("pi");
65 compareArray_real("pi", pi, ref_pi, k);
66 free(pi);
67 free(ref_pi);
68
69 Real ref_llh = read_real("llh");
70 compareArray_real("llh", &llh, &ref_llh, 1);
71
72 Real* ref_S = readArray_real("S");
73 compareArray_real("S", S, ref_S, p*m*k);
74 free(S);
75 free(ref_S);
76
77 int* ref_affec = readArray_int("affec");
78 compareArray_int("affec", affec, ref_affec, n);
79 free(affec);
80 free(ref_affec);
81
82 return 0;
83 }