add tests for a^x

coin-or · Jan 8, 2025 · 697979b · 697979b
1 parent 616b728
commit 697979b
Show file tree

Hide file tree

Showing 2 changed files with 143 additions and 0 deletions.
diff --git a/ADOL-C/boost-test/uni5_for/hov_forward.cpp b/ADOL-C/boost-test/uni5_for/hov_forward.cpp
@@ -2574,4 +2574,75 @@ BOOST_AUTO_TEST_CASE(Atan2Operator_HOV_Forward) {
   myfree3(X);
   myfree3(Y);
 }
+
+BOOST_AUTO_TEST_CASE(Pow_Operator_HOV_Forward_1) {
+  const int16_t tag = 0;
+  const size_t dim_out = 1;
+  const size_t dim_in = 1;
+  const size_t degree = 3;
+  const size_t num_dirs = 2;
+  std::vector<double> in{0.2};
+  std::vector<adouble> indep(dim_in);
+  std::vector<double> out(dim_out);
+
+  // x^y
+  trace_on(tag);
+  indep[0] <<= in[0];
+  adouble dep = pow(indep[0], 3.2);
+  dep >>= out[0];
+  trace_off();
+
+  double ***X = myalloc3(dim_in, num_dirs, degree);
+  double ***Y = myalloc3(dim_out, num_dirs, degree);
+
+  X[0][0][0] = 1.0;
+  X[0][0][1] = -1.0;
+  X[0][0][2] = 1.1;
+
+  X[0][1][0] = 2.0;
+  X[0][1][1] = -2.1;
+  X[0][1][2] = -2.0;
+
+  std::vector<double> test_in{0.2};
+
+  // x^y
+  double test_out = std::pow(test_in[0], 3.2);
+
+  hov_forward(tag, dim_out, dim_in, degree, num_dirs, test_in.data(), X,
+              out.data(), Y);
+
+  BOOST_TEST(out[0] == test_out, tt::tolerance(tol));
+
+  // first derivative
+  BOOST_TEST(Y[0][0][0] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][0] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][0],
+             tt::tolerance(tol));
+
+  // second derivative
+  BOOST_TEST(Y[0][0][1] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][1] +
+                               1.0 / 2.0 * std::pow(test_in[0], 1.2) * 7.04 *
+                                   X[0][0][0] * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][1] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][1] +
+                               1.0 / 2.0 * std::pow(test_in[0], 1.2) * 7.04 *
+                                   X[0][1][0] * X[0][1][0],
+             tt::tolerance(tol));
+
+  // third derivative
+  BOOST_TEST(Y[0][0][2] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][2] +
+                               std::pow(test_in[0], 1.2) * 7.04 * X[0][0][0] *
+                                   X[0][0][1] +
+                               1.0 / 6.0 * std::pow(test_in[0], 0.2) * 8.448 *
+                                   X[0][0][0] * X[0][0][0] * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][2] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][2] +
+                               std::pow(test_in[0], 1.2) * 7.04 * X[0][1][0] *
+                                   X[0][1][1] +
+                               1.0 / 6.0 * std::pow(test_in[0], 0.2) * 8.448 *
+                                   X[0][1][0] * X[0][1][0] * X[0][1][0],
+             tt::tolerance(tol));
+  myfree3(X);
+  myfree3(Y);
+}
 BOOST_AUTO_TEST_SUITE_END()
diff --git a/ADOL-C/boost-test/uni5_for/hov_wk_forward.cpp b/ADOL-C/boost-test/uni5_for/hov_wk_forward.cpp
@@ -2601,4 +2601,76 @@ BOOST_AUTO_TEST_CASE(Atan2Operator_HOV_Forward) {
   myfree3(X);
   myfree3(Y);
 }
+
+BOOST_AUTO_TEST_CASE(Pow_Operator_HOV_Forward_1) {
+  const int16_t tag = 0;
+  const size_t dim_out = 1;
+  const size_t dim_in = 1;
+  const size_t degree = 3;
+  const size_t num_dirs = 2;
+  const short keep = 1;
+  std::vector<double> in{0.2};
+  std::vector<adouble> indep(dim_in);
+  std::vector<double> out(dim_out);
+
+  // x^y
+  trace_on(tag);
+  indep[0] <<= in[0];
+  adouble dep = pow(indep[0], 3.2);
+  dep >>= out[0];
+  trace_off();
+
+  double ***X = myalloc3(dim_in, num_dirs, degree);
+  double ***Y = myalloc3(dim_out, num_dirs, degree);
+
+  X[0][0][0] = 1.0;
+  X[0][0][1] = -1.0;
+  X[0][0][2] = 1.1;
+
+  X[0][1][0] = 2.0;
+  X[0][1][1] = -2.1;
+  X[0][1][2] = -2.0;
+
+  std::vector<double> test_in{0.2};
+
+  // x^y
+  double test_out = std::pow(test_in[0], 3.2);
+
+  hov_wk_forward(tag, dim_out, dim_in, degree, keep, num_dirs, test_in.data(),
+                 X, out.data(), Y);
+
+  BOOST_TEST(out[0] == test_out, tt::tolerance(tol));
+
+  // first derivative
+  BOOST_TEST(Y[0][0][0] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][0] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][0],
+             tt::tolerance(tol));
+
+  // second derivative
+  BOOST_TEST(Y[0][0][1] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][1] +
+                               1.0 / 2.0 * std::pow(test_in[0], 1.2) * 7.04 *
+                                   X[0][0][0] * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][1] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][1] +
+                               1.0 / 2.0 * std::pow(test_in[0], 1.2) * 7.04 *
+                                   X[0][1][0] * X[0][1][0],
+             tt::tolerance(tol));
+
+  // third derivative
+  BOOST_TEST(Y[0][0][2] == std::pow(test_in[0], 2.2) * 3.2 * X[0][0][2] +
+                               std::pow(test_in[0], 1.2) * 7.04 * X[0][0][0] *
+                                   X[0][0][1] +
+                               1.0 / 6.0 * std::pow(test_in[0], 0.2) * 8.448 *
+                                   X[0][0][0] * X[0][0][0] * X[0][0][0],
+             tt::tolerance(tol));
+  BOOST_TEST(Y[0][1][2] == std::pow(test_in[0], 2.2) * 3.2 * X[0][1][2] +
+                               std::pow(test_in[0], 1.2) * 7.04 * X[0][1][0] *
+                                   X[0][1][1] +
+                               1.0 / 6.0 * std::pow(test_in[0], 0.2) * 8.448 *
+                                   X[0][1][0] * X[0][1][0] * X[0][1][0],
+             tt::tolerance(tol));
+  myfree3(X);
+  myfree3(Y);
+}
 BOOST_AUTO_TEST_SUITE_END()