test with matrix arguments

Author: dance858

tests/all_tests.c

@@ -262,6 +262,9 @@ int main(void)
     mu_run_test(test_wsum_hess_transpose, tests_run);
     mu_run_test(test_wsum_hess_exp_sum, tests_run);
     mu_run_test(test_wsum_hess_exp_sum_mult, tests_run);
+    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);
 
     printf("\n--- Utility Tests ---\n");
     mu_run_test(test_cblas_ddot, tests_run);

tests/wsum_hess/test_chain_rule_wsum_hess.h

@@ -37,3 +37,64 @@ const char *test_wsum_hess_exp_sum_mult(void)
     free_expr(exp_sum_xy);
     return 0;
 }
+
+const char *test_wsum_hess_exp_sum_matmul(void)
+{
+    /* exp(sum(X @ Y)) where X is 2x3, Y is 3x2
+     * n_vars = 6 + 6 = 12 */
+    double u_vals[12] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0.5, 1.5, 2.5, 0.1, 0.2, 0.3};
+    double w = 1.0;
+
+    expr *X = new_variable(2, 3, 0, 12);
+    expr *Y = new_variable(3, 2, 6, 12);
+    expr *XY = new_matmul(X, Y);
+    expr *sum_XY = new_sum(XY, -1);
+    expr *exp_sum_XY = new_exp(sum_XY);
+
+    mu_assert("check_wsum_hess failed",
+              check_wsum_hess(exp_sum_XY, u_vals, &w, NUMERICAL_DIFF_DEFAULT_H));
+
+    free_expr(exp_sum_XY);
+    return 0;
+}
+
+const char *test_wsum_hess_sin_sum_axis0_matmul(void)
+{
+    /* sin(sum(X @ Y, axis=0)) where X is 2x3, Y is 3x2
+     * X@Y is 2x2, sum(axis=0) gives 1x2, sin gives 1x2
+     * n_vars = 6 + 6 = 12 */
+    double u_vals[12] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0.5, 1.5, 2.5, 0.1, 0.2, 0.3};
+    double w[2] = {1.0, 1.0};
+
+    expr *X = new_variable(2, 3, 0, 12);
+    expr *Y = new_variable(3, 2, 6, 12);
+    expr *XY = new_matmul(X, Y);
+    expr *sum_XY = new_sum(XY, 0);
+    expr *sin_sum_XY = new_sin(sum_XY);
+
+    mu_assert("check_wsum_hess failed",
+              check_wsum_hess(sin_sum_XY, u_vals, w, NUMERICAL_DIFF_DEFAULT_H));
+
+    free_expr(sin_sum_XY);
+    return 0;
+}
+
+const char *test_wsum_hess_logistic_sum_axis0_matmul(void)
+{
+    /* logistic(sum(X @ Y, axis=0)) where X is 2x3, Y is 3x2
+     * n_vars = 6 + 6 = 12 */
+    double u_vals[12] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0.5, 1.5, 2.5, 0.1, 0.2, 0.3};
+    double w[2] = {1.0, 1.0};
+
+    expr *X = new_variable(2, 3, 0, 12);
+    expr *Y = new_variable(3, 2, 6, 12);
+    expr *XY = new_matmul(X, Y);
+    expr *sum_XY = new_sum(XY, 0);
+    expr *logistic_sum_XY = new_logistic(sum_XY);
+
+    mu_assert("check_wsum_hess failed",
+              check_wsum_hess(logistic_sum_XY, u_vals, w, NUMERICAL_DIFF_DEFAULT_H));
+
+    free_expr(logistic_sum_XY);
+    return 0;
+}