Merge pull request #11 from DCM-UPB/multidim_estimator_unittest

Added unittest 2

Author: Ithanil

.gitignore

@@ -62,6 +62,7 @@ DerivedData
 lib*
 config.sh
 *exe*
+vgcore*
 
 src/*.o
 

debug/unittests/README.md

@@ -5,3 +5,7 @@
 ## Unit Test 1
 
 `ut1/`: check that calling the integrator again without doing the findMRT2step and initialDecorrelation gives the same results.
+
+## Unit Test 2
+
+`ut2/`: Like ut1, but with one one-dimensional and one three-dimensional observable.

debug/unittests/ut1/main.cpp

@@ -76,7 +76,7 @@ int main(){
     delete pdf;
     delete obs;
     delete mci;
-    delete x;
+    delete [] x;
     delete average;
     delete error;
 

debug/unittests/ut2/main.cpp

@@ -0,0 +1,104 @@
+#include "MCIntegrator.hpp"
+#include "MCISamplingFunctionInterface.hpp"
+
+#include <iostream>
+#include <math.h>
+#include <assert.h>
+
+using namespace std;
+
+
+
+class ThreeDimGaussianPDF: public MCISamplingFunctionInterface{
+public:
+    ThreeDimGaussianPDF(): MCISamplingFunctionInterface(3, 1){}
+    ~ThreeDimGaussianPDF(){}
+
+    void samplingFunction(const double *in, double * protovalues){
+        protovalues[0] = (in[0]*in[0]) + (in[1]*in[1]) + (in[2]*in[2]);
+    }
+
+
+    double getAcceptance(const double * protoold, const double * protonew){
+        return exp(-protonew[0]+protoold[0]);
+    }
+
+};
+
+class XSquared: public MCIObservableFunctionInterface{
+public:
+    XSquared(): MCIObservableFunctionInterface(3, 1){}
+    ~XSquared(){}
+
+    void observableFunction(const double * in, double * out){
+        out[0] = in[0] * in[0];
+    }
+};
+
+
+class XYZSquared: public MCIObservableFunctionInterface{
+public:
+    XYZSquared(): MCIObservableFunctionInterface(3, 3){}
+    ~XYZSquared(){}
+
+    void observableFunction(const double * in, double * out){
+        out[0] = in[0] * in[0];
+        out[1] = in[1] * in[1];
+        out[2] = in[2] * in[2];
+    }
+};
+
+
+
+int main(){
+    const long NMC = 10000;
+    const double CORRECT_RESULT = 0.5;
+
+    ThreeDimGaussianPDF * pdf = new ThreeDimGaussianPDF();
+    XSquared * obs1d = new XSquared();
+    XYZSquared * obs3d = new XYZSquared();
+
+    MCI * mci = new MCI(3);
+    mci->setSeed(5649871);
+    mci->addSamplingFunction(pdf);
+    mci->addObservable(obs1d);
+    mci->addObservable(obs3d);
+    // the integral should provide 0.5 as answer!
+
+    double * x = new double[3];
+    x[0] = 5.; x[1] = -5.; x[2] = 10.;
+
+    double * average = new double[4];
+    double * error = new double[4];
+
+    // this integral will give a wrong answer! This is because the starting point is very bad and initialDecorrelation is skipped (as well as the MRT2step automatic setting)
+    mci->setX(x);
+    mci->integrate(NMC, average, error, false, false);
+    for (int i=0; i<mci->getNObsDim(); ++i) {
+        assert( abs(average[i]-CORRECT_RESULT) > 2.*error[i] );
+    }
+
+    // this integral, instead, will provide the right answer
+    mci->setX(x);
+    mci->integrate(NMC, average, error);
+    for (int i=0; i<mci->getNObsDim(); ++i) {
+        assert( abs(average[i]-CORRECT_RESULT) < 2.*error[i] );
+    }
+
+    // now, doing an integral without finding again the MRT2step and doing the initialDecorrelation will also result in a correct result
+    mci->integrate(NMC, average, error, false, false);
+    for (int i=0; i<mci->getNObsDim(); ++i) {
+        assert( abs(average[i]-CORRECT_RESULT) < 2.*error[i] );
+    }
+
+
+    delete pdf;
+    delete obs1d;
+    delete obs3d;
+    delete mci;
+    delete [] x;
+    delete [] average;
+    delete [] error;
+
+    return 0;
+}