#include "test_helpers.h"
#include "gr_mat.h"
#include "gr_poly.h"
FLINT_DLL extern gr_static_method_table _ca_methods;
TEST_GR_FUNCTION_START(gr_mat_solve_lu, state, count_success, count_unable, count_domain)
{
slong iter;
for (iter = 0; iter < 1000; iter++)
{
gr_ctx_t ctx;
gr_mat_t A, X, B, AX;
slong n, c;
int status = GR_SUCCESS;
n = n_randint(state, 6);
c = n_randint(state, 6);
gr_ctx_init_random_commutative_ring(ctx, state);
while (ctx->methods == _ca_methods)
{
gr_ctx_clear(ctx);
gr_ctx_init_random(ctx, state);
}
gr_mat_init(A, n, n, ctx);
gr_mat_init(B, n, c, ctx);
gr_mat_init(X, n, c, ctx);
gr_mat_init(AX, n, c, ctx);
status |= gr_mat_randtest(A, state, ctx);
status |= gr_mat_randtest(X, state, ctx);
status |= gr_mat_randtest(B, state, ctx);
status |= gr_mat_nonsingular_solve_lu(X, A, B, ctx);
if (status == GR_SUCCESS)
{
status |= gr_mat_mul(AX, A, X, ctx);
if (status == GR_SUCCESS && gr_mat_equal(AX, B, ctx) == T_FALSE)
{
flint_printf("FAIL\n");
gr_ctx_println(ctx);
flint_printf("A = \n"); gr_mat_print(A, ctx); flint_printf("\n\n");
flint_printf("B = \n"); gr_mat_print(B, ctx); flint_printf("\n\n");
flint_printf("X = \n"); gr_mat_print(X, ctx); flint_printf("\n\n");
flint_printf("AX = \n"); gr_mat_print(AX, ctx); flint_printf("\n\n");
flint_abort();
}
}
else if ((status & GR_DOMAIN) && !(status & GR_UNABLE) && gr_ctx_is_field(ctx) == T_TRUE)
{
gr_ptr det;
int status2;
det = gr_heap_init(ctx);
status2 = gr_mat_det_berkowitz(det, A, ctx);
if (status2 == GR_SUCCESS && gr_is_invertible(det, ctx) == T_TRUE)
{
flint_printf("FAIL (singular)\n\n");
gr_ctx_println(ctx);
flint_printf("A = "); gr_mat_print(A, ctx); flint_printf("\n");
flint_printf("det = "); gr_print(det, ctx); flint_printf("\n");
flint_abort();
}
gr_heap_clear(det, ctx);
}
count_success += (status == GR_SUCCESS);
count_domain += ((status & GR_DOMAIN) != 0);
count_unable += ((status & GR_UNABLE) != 0);
gr_mat_clear(A, ctx);
gr_mat_clear(B, ctx);
gr_mat_clear(X, ctx);
gr_mat_clear(AX, ctx);
gr_ctx_clear(ctx);
}
TEST_GR_FUNCTION_END(state, count_success, count_unable, count_domain);
}