src/test/test.selectiontotale.c

   1 #include "selectiontotale.h"
   2 #include "test_utils.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 }