[WIP] more tests (#62)

* more tests

* run formatter

Author: dance858

src/affine/trace.c

@@ -102,7 +102,7 @@ static void wsum_hess_init(expr *node)
 
     /* initialize child's hessian */
     x->wsum_hess_init(x);
-  
+
     node->work->dwork = (double *) calloc(x->size, sizeof(double));
 
     /* We copy over the sparsity pattern from the child. This also includes the

tests/all_tests.c

@@ -133,6 +133,7 @@ int main(void)
     mu_run_test(test_jacobian_composite_exp, tests_run);
     mu_run_test(test_jacobian_exp_sum, tests_run);
     mu_run_test(test_jacobian_exp_sum_mult, tests_run);
+    mu_run_test(test_jacobian_sin_cos, tests_run);
     mu_run_test(test_jacobian_composite_exp_add, tests_run);
     mu_run_test(test_jacobian_const_scalar_mult_log_vector, tests_run);
     mu_run_test(test_jacobian_const_scalar_mult_log_matrix, tests_run);
@@ -265,6 +266,7 @@ int main(void)
     mu_run_test(test_wsum_hess_exp_sum_matmul, tests_run);
     mu_run_test(test_wsum_hess_sin_sum_axis0_matmul, tests_run);
     mu_run_test(test_wsum_hess_logistic_sum_axis0_matmul, tests_run);
+    mu_run_test(test_wsum_hess_sin_cos, tests_run);
 
     printf("\n--- Utility Tests ---\n");
     mu_run_test(test_cblas_ddot, tests_run);

tests/jacobian_tests/test_chain_rule_jacobian.h

@@ -35,3 +35,18 @@ const char *test_jacobian_exp_sum_mult(void)
     free_expr(exp_sum_xy);
     return 0;
 }
+
+const char *test_jacobian_sin_cos(void)
+{
+    double u_vals[5] = {0.5, 1.0, 1.5, 2.0, 2.5};
+
+    expr *x = new_variable(5, 1, 0, 5);
+    expr *cos_x = new_cos(x);
+    expr *sin_cos_x = new_sin(cos_x);
+
+    mu_assert("check_jacobian failed",
+              check_jacobian(sin_cos_x, u_vals, NUMERICAL_DIFF_DEFAULT_H));
+
+    free_expr(sin_cos_x);
+    return 0;
+}

tests/wsum_hess/test_chain_rule_wsum_hess.h

@@ -98,3 +98,19 @@ const char *test_wsum_hess_logistic_sum_axis0_matmul(void)
     free_expr(logistic_sum_XY);
     return 0;
 }
+
+const char *test_wsum_hess_sin_cos(void)
+{
+    double u_vals[5] = {0.5, 1.0, 1.5, 2.0, 2.5};
+    double w[5] = {1.0, 2.0, 3.0, 4.0, 5.0};
+
+    expr *x = new_variable(5, 1, 0, 5);
+    expr *cos_x = new_cos(x);
+    expr *sin_cos_x = new_sin(cos_x);
+
+    mu_assert("check_wsum_hess failed",
+              check_wsum_hess(sin_cos_x, u_vals, w, NUMERICAL_DIFF_DEFAULT_H));
+
+    free_expr(sin_cos_x);
+    return 0;
+}