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 }