added test case for polynomial of total degree 2

mluq/src/problems/IStochasticProblem.cpp

@@ -23,7 +23,7 @@ CreateStochasticDummyProblem(const std::string &problemName, const Meshes &meshe
     return new SparseGrid1DGeneratorProblem(meshes);
   if (problemName == "SparseGrid2DGeneratorProblem")
     return new SparseGrid2DGeneratorProblem(meshes);
-  if (problemName == "SparseGrid2DPolynomialDegree2Problem")
-    return new SparseGrid2DPolynomialDegree2Problem(meshes);
+  if (problemName == "SparseGrid2DPolyDeg2Problem")
+    return new SparseGrid2DPolyDeg2Problem(meshes);
   Exit(problemName + " not found")
 }
\ No newline at end of file

mluq/src/problems/IStochasticProblem.hpp

@@ -249,22 +249,22 @@ public:
   }
 };
 
-class SparseGrid2DPolynomialDegree2Problem : public SparseGridGeneratorProblem {
+class SparseGrid2DPolyDeg2Problem : public SparseGridGeneratorProblem {
 public:
-  explicit SparseGrid2DPolynomialDegree2Problem(const Meshes &meshes) :
-    SparseGridGeneratorProblem(meshes, SparseGridGenerator(meshes, 2, 0, 1)) {}
+  explicit SparseGrid2DPolyDeg2Problem(const Meshes &meshes) :
+    SparseGridGeneratorProblem(meshes, SparseGridGenerator(meshes, 2, 0, 4)) {}
 
   double FunctionEvaluation() override {
     RVector sample = this->generator.EvalSample();
-    return sample[0] * sample[1];
+    return 1 + sample[0] + pow(sample[0],2) + sample[1] + pow(sample[1],2) + sample[0] * sample[1];
   }
 
   string Name() const override {
-    return "SparseGrid2DPolynomialDegree2Problem";
+    return "SparseGrid2DPolyDeg2Problem";
   }
 };
 
-// Todo find more test cases in tasmanina examples and write sparse grid problems
+// Todo find more test cases in tasmanian examples and write sparse grid problems
 
 StochasticDummyProblem *
 CreateStochasticDummyProblem(const std::string &problemName, const Meshes &meshes);

mluq/tests/estimators/TestStochasticCollocation.cpp

@@ -9,8 +9,8 @@ INSTANTIATE_TEST_SUITE_P(
   TestStochasticCollocation, TestStochasticCollocationWithoutEpsilon, Values(
   TestParams{"SparseGrid2DGeneratorProblem", "FunctionEvaluation",
              "DummyPDESolver", 2.513723354063905},
-  TestParams{"SparseGrid2DPolynomialDegree2Problem", "FunctionEvaluation",
-             "DummyPDESolver", 0.0}
+  TestParams{"SparseGrid2DPolyDeg2Problem", "FunctionEvaluation",
+             "DummyPDESolver", 6.666666666666667}
 
   // Todo add more test cases
 ));