From 6cad9f3d1edbdfd0938cd0eb1fc8cc2de6319ed3 Mon Sep 17 00:00:00 2001
From: "petro.zarytskyi" <petro.zarytskyi@gmail.com>
Date: Fri, 25 Oct 2024 20:06:22 +0200
Subject: [PATCH] Add tests for new features

---
 test/Jacobian/Jacobian.C | 62 ++++++++++++++++++++++++++++++++++++----
 1 file changed, 56 insertions(+), 6 deletions(-)

diff --git a/test/Jacobian/Jacobian.C b/test/Jacobian/Jacobian.C
index 3aa8e4f76..5f4ad38b6 100644
--- a/test/Jacobian/Jacobian.C
+++ b/test/Jacobian/Jacobian.C
@@ -124,11 +124,50 @@ void f_5(float a, double output[]){
 // CHECK-NEXT:     output[0] = a * a;
 // CHECK-NEXT: }
 
+void f_6(double a[3], double b, double output[]) {
+    output[0] = a[0] * a[0] * a[0];
+    output[1] = a[0] * a[0] * a[0] + b * b * b;
+    output[2] = 2 * (a[0] + b);
+}
+
+// CHECK: void f_6_jac(double a[3], double b, double output[], clad::matrix<double> *_d_vector_a, clad::matrix<double> *_d_vector_output) {
+// CHECK-NEXT:     unsigned long indepVarCount = _d_vector_a->rows() + _d_vector_output->rows() + {{1U|1UL|1ULL}};
+// CHECK-NEXT:     clad::array<double> _d_vector_b = clad::one_hot_vector(indepVarCount, _d_vector_a->rows());
+// CHECK-NEXT:     *_d_vector_a = clad::identity_matrix(_d_vector_a->rows(), indepVarCount, {{0U|0UL|0ULL}});
+// CHECK-NEXT:     *_d_vector_output = clad::identity_matrix(_d_vector_output->rows(), indepVarCount, _d_vector_a->rows() + {{1U|1UL|1ULL}});
+// CHECK-NEXT:     double _t0 = a[0] * a[0];
+// CHECK-NEXT:     *_d_vector_output[0] = ((*_d_vector_a[0]) * a[0] + a[0] * (*_d_vector_a[0])) * a[0] + _t0 * (*_d_vector_a[0]);
+// CHECK-NEXT:     output[0] = _t0 * a[0];
+// CHECK-NEXT:     double _t1 = a[0] * a[0];
+// CHECK-NEXT:     double _t2 = b * b;
+// CHECK-NEXT:     *_d_vector_output[1] = ((*_d_vector_a[0]) * a[0] + a[0] * (*_d_vector_a[0])) * a[0] + _t1 * (*_d_vector_a[0]) + (_d_vector_b * b + b * _d_vector_b) * b + _t2 * _d_vector_b;
+// CHECK-NEXT:     output[1] = _t1 * a[0] + _t2 * b;
+// CHECK-NEXT:     double _t3 = (a[0] + b);
+// CHECK-NEXT:     *_d_vector_output[2] = 0 * _t3 + 2 * (*_d_vector_a[0] + _d_vector_b);
+// CHECK-NEXT:     output[2] = 2 * _t3;
+// CHECK-NEXT: }
+
+void f_7(double a, double& b, double&c) {
+    b = 3 * a;
+    c = -5 * b;
+}
 
-#define TEST(F, x, y, z) { \
+// CHECK: void f_7_jac(double a, double &b, double &c, clad::array<double> *_d_vector_b, clad::array<double> *_d_vector_c) {
+// CHECK-NEXT:     unsigned long indepVarCount = {{3U|3UL|3ULL}};
+// CHECK-NEXT:     clad::array<double> _d_vector_a = clad::one_hot_vector(indepVarCount, {{0U|0UL|0ULL}});
+// CHECK-NEXT:     *_d_vector_b = clad::one_hot_vector(indepVarCount, {{1U|1UL|1ULL}});
+// CHECK-NEXT:     *_d_vector_c = clad::one_hot_vector(indepVarCount, {{2U|2UL|2ULL}});
+// CHECK-NEXT:     *_d_vector_b = 0 * a + 3 * _d_vector_a;
+// CHECK-NEXT:     b = 3 * a;
+// CHECK-NEXT:     *_d_vector_c = -0 * b + -5 * *_d_vector_b;
+// CHECK-NEXT:     c = -5 * b;
+// CHECK-NEXT: }
+
+
+#define TEST(F, ...) { \
   outputarr[0] = 0; outputarr[1] = 1; outputarr[2] = 0;\
   auto j = clad::jacobian(F);\
-  j.execute(x, y, z, outputarr, &result);\
+  j.execute(__VA_ARGS__, &result);\
   printf("Result is = {%.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f, %.2f}\n",\
   result[0][0], result[0][1], result[0][2],\
   result[1][0], result[1][1], result[1][2],\
@@ -147,15 +186,26 @@ void f_5(float a, double output[]){
 int main() {
   clad::matrix<double> result (3, 3);
   double outputarr[9];
-  TEST(f_1, 1, 2, 3); // CHECK-EXEC: Result is = {3.00, 0.00, 0.00, 3.00, 12.00, 0.00, -2.00, 0.00, 60.00}
-  TEST(f_3, 1, 2, 3); // CHECK-EXEC: Result is = {22.69, 0.00, 0.00, 0.00, -17.48, 0.00, 0.00, 0.00, -41.58}
-  TEST(f_4, 1, 2, 3); // CHECK-EXEC: Result is = {84.00, 42.00, 0.00, 0.00, 126.00, 84.00, 126.00, 0.00, 42.00}
+  TEST(f_1, 1, 2, 3, outputarr); // CHECK-EXEC: Result is = {3.00, 0.00, 0.00, 3.00, 12.00, 0.00, -2.00, 0.00, 60.00}
+  TEST(f_3, 1, 2, 3, outputarr); // CHECK-EXEC: Result is = {22.69, 0.00, 0.00, 0.00, -17.48, 0.00, 0.00, 0.00, -41.58}
+  TEST(f_4, 1, 2, 3, outputarr); // CHECK-EXEC: Result is = {84.00, 42.00, 0.00, 0.00, 126.00, 84.00, 126.00, 0.00, 42.00}
   TEST_F_1_SINGLE_PARAM(1, 2, 3); // CHECK-EXEC: Result is = {3.00, 3.00, -2.00}
 
   auto df5 = clad::jacobian(f_5);
   result[0] = 0; result[1] = 0;
   df5.execute(3, outputarr, &result);
-  printf("Result is = {%.2f, %.2f}", result[0][0], result[1][0]); // CHECK-EXEC: Result is = {6.00, 1.00}
+  printf("Result is = {%.2f, %.2f}\n", result[0][0], result[1][0]); // CHECK-EXEC: Result is = {6.00, 1.00}
+  
+  double a[] = {3, 1};
+  clad::matrix<double> da (2, 5);
+  TEST(f_6, a, 2, outputarr, &da); // CHECK-EXEC: Result is = {27.00, 0.00, 0.00, 27.00, 0.00, 12.00, 2.00, 0.00, 2.00}
+
+  auto df7 = clad::jacobian(f_7);
+  clad::array<double> db, dc;
+  double a7=4, b=5, c=6;
+  df7.execute(a7, b, c, &db, &dc);
+  printf("Result is = {%.2f, %.2f}\n", db[0], dc[0]); // CHECK-EXEC: Result is = {3.00, -15.00}
+  
 }
 
 // CHECK: clad::ValueAndPushforward<double, clad::array<double> > multiply_vector_pushforward(double x, double y, clad::array<double> _d_x, clad::array<double> _d_y) {