blob: 10ff138b9d98879073f81f6d1bac09fa0d53a143 [file] [log] [blame]
Austin Schuh9049e202022-02-20 17:34:16 -08001#include "osqp.h" // OSQP API
2#include "osqp_tester.h" // Basic testing script header
3
4void test_demo_solve()
5{
6 // Load problem data
7 c_float P_x[3] = { 4.0, 1.0, 2.0, };
8 c_int P_nnz = 3;
9 c_int P_i[3] = { 0, 0, 1, };
10 c_int P_p[3] = { 0, 1, 3, };
11 c_float q[2] = { 1.0, 1.0, };
12 c_float A_x[4] = { 1.0, 1.0, 1.0, 1.0, };
13 c_int A_nnz = 4;
14 c_int A_i[4] = { 0, 1, 0, 2, };
15 c_int A_p[3] = { 0, 2, 4, };
16 c_float l[3] = { 1.0, 0.0, 0.0, };
17 c_float u[3] = { 1.0, 0.7, 0.7, };
18 c_int n = 2;
19 c_int m = 3;
20
21 c_int exitflag;
22
23 // Problem settings
24 OSQPSettings *settings = (OSQPSettings *)c_malloc(sizeof(OSQPSettings));
25
26 // Structures
27 OSQPWorkspace *work; // Workspace
28 OSQPData *data; // OSQPData
29
30 // Populate data
31 data = (OSQPData *)c_malloc(sizeof(OSQPData));
32 data->n = n;
33 data->m = m;
34 data->P = csc_matrix(data->n, data->n, P_nnz, P_x, P_i, P_p);
35 data->q = q;
36 data->A = csc_matrix(data->m, data->n, A_nnz, A_x, A_i, A_p);
37 data->l = l;
38 data->u = u;
39
40 // Define solver settings as default
41 osqp_set_default_settings(settings);
42
43 // Setup workspace
44 exitflag = osqp_setup(&work, data, settings);
45
46 // Setup correct
47 mu_assert("Demo test solve: Setup error!", exitflag == 0);
48
49 // Solve Problem
50 osqp_solve(work);
51
52 // Compare solver statuses
53 mu_assert("Demo test solve: Error in solver status!",
54 work->info->status_val == OSQP_SOLVED);
55
56 // Clean workspace
57 osqp_cleanup(work);
58 c_free(data->A);
59 c_free(data->P);
60 c_free(data);
61 c_free(settings);
62}