Merge pull request #1 from dance858/adding-tests

Adds tests to begin familiarizing with the diff engine.

Author: dance858

tests/all_tests.c

@@ -53,6 +53,7 @@ int main(void)
 
     printf("\n--- Jacobian Tests ---\n");
     mu_run_test(test_jacobian_log, tests_run);
+    mu_run_test(test_jacobian_log_matrix, tests_run);
     mu_run_test(test_jacobian_composite_log, tests_run);
     mu_run_test(test_jacobian_composite_log_add, tests_run);
     mu_run_test(test_jacobian_rel_entr_vector_args_1, tests_run);

tests/jacobian_tests/test_log.h

@@ -24,3 +24,22 @@ const char *test_jacobian_log()
     free_expr(u);
     return 0;
 }
+
+const char *test_jacobian_log_matrix()
+{
+    double u_vals[7] = {0.0, 0.0, 0.0, 1.0, 2.0, 4.0, 5.0};
+    double expected_Ax[4] = {1.0, 0.5, 0.25, 0.2};
+    int expected_Ap[5] = {0, 1, 2, 3, 4};
+    int expected_Ai[4] = {3, 4, 5, 6};
+    expr *u = new_variable(2, 2, 3, 7);
+    expr *log_node = new_log(u);
+    log_node->forward(log_node, u_vals);
+    log_node->jacobian_init(log_node);
+    log_node->eval_jacobian(log_node);
+    mu_assert("vals fail", cmp_double_array(log_node->jacobian->x, expected_Ax, 4));
+    mu_assert("rows fail", cmp_int_array(log_node->jacobian->p, expected_Ap, 5));
+    mu_assert("cols fail", cmp_int_array(log_node->jacobian->i, expected_Ai, 4));
+    free_expr(log_node);
+    free_expr(u);
+    return 0;
+}