#include "math/lp/nla_intervals.h"
#include "math/lp/nla_core.h"
#include "math/lp/lar_solver.h"
#include "util/rational.h"
#include "util/rlimit.h"
#include <iostream>
namespace nla {
void test_nla_intervals_basic() {
std::cout << "test_nla_intervals_basic\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar y = s.add_var(1, true);
lpvar xy = s.add_var(2, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(y);
nla_solver.add_monic(xy, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(1));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(3));
s.add_var_bound(y, lp::lconstraint_kind::GE, rational(2));
s.add_var_bound(y, lp::lconstraint_kind::LE, rational(4));
VERIFY(true); }
void test_nla_intervals_negative() {
std::cout << "test_nla_intervals_negative\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar y = s.add_var(1, true);
lpvar xy = s.add_var(2, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(y);
nla_solver.add_monic(xy, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(-3));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(-1));
s.add_var_bound(y, lp::lconstraint_kind::GE, rational(2));
s.add_var_bound(y, lp::lconstraint_kind::LE, rational(4));
VERIFY(true); }
void test_nla_intervals_zero_crossing() {
std::cout << "test_nla_intervals_zero_crossing\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar y = s.add_var(1, true);
lpvar xy = s.add_var(2, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(y);
nla_solver.add_monic(xy, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(-2));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(3));
s.add_var_bound(y, lp::lconstraint_kind::GE, rational(1));
s.add_var_bound(y, lp::lconstraint_kind::LE, rational(4));
VERIFY(true); }
void test_nla_intervals_power() {
std::cout << "test_nla_intervals_power\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar x_squared = s.add_var(1, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(x);
nla_solver.add_monic(x_squared, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(-3));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(2));
VERIFY(true); }
void test_nla_intervals_mixed_signs() {
std::cout << "test_nla_intervals_mixed_signs\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar y = s.add_var(1, true);
lpvar z = s.add_var(2, true);
lpvar xyz = s.add_var(3, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(y);
vars.push_back(z);
nla_solver.add_monic(xyz, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(-1));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(1));
s.add_var_bound(y, lp::lconstraint_kind::GE, rational(-2));
s.add_var_bound(y, lp::lconstraint_kind::LE, rational(2));
s.add_var_bound(z, lp::lconstraint_kind::GE, rational(1));
s.add_var_bound(z, lp::lconstraint_kind::LE, rational(3));
VERIFY(true); }
void test_nla_intervals_fractional() {
std::cout << "test_nla_intervals_fractional\n";
reslimit rl;
params_ref p;
lp::lar_solver s;
lpvar x = s.add_var(0, true);
lpvar y = s.add_var(1, true);
lpvar xy = s.add_var(2, true);
nla::core nla_solver(s, p, rl);
vector<lpvar> vars;
vars.push_back(x);
vars.push_back(y);
nla_solver.add_monic(xy, vars.size(), vars.begin());
s.add_var_bound(x, lp::lconstraint_kind::GE, rational(1, 2));
s.add_var_bound(x, lp::lconstraint_kind::LE, rational(3, 2));
s.add_var_bound(y, lp::lconstraint_kind::GE, rational(5, 2));
s.add_var_bound(y, lp::lconstraint_kind::LE, rational(7, 2));
VERIFY(true); }
void test_fetch_normalized_term_column() {
std::cout << "test_fetch_normalized_term_column\n";
lp::lar_solver s;
lpvar x = s.add_var(0, true); lpvar y = s.add_var(1, true); lpvar z = s.add_var(2, true);
lp::lar_term t;
t.add_monomial(rational(2), x);
t.add_monomial(rational(3), y);
s.add_term(t.coeffs_as_vector(), UINT_MAX);
s.register_existing_terms();
lp::lar_term query;
query.add_monomial(rational(2), x);
query.add_monomial(rational(3), y);
lp::mpq a;
lp::lar_term norm_query = query.get_normalized_by_min_var(a);
std::pair<lp::mpq, lpvar> result;
bool found = s.fetch_normalized_term_column(norm_query, result);
VERIFY(found);
std::cout << " round-trip lookup: " << (found ? "PASS" : "FAIL") << "\n";
lp::lar_term query_rev;
query_rev.add_monomial(rational(3), y);
query_rev.add_monomial(rational(2), x);
lp::lar_term norm_rev = query_rev.get_normalized_by_min_var(a);
bool found_rev = s.fetch_normalized_term_column(norm_rev, result);
VERIFY(found_rev);
std::cout << " reverse-order lookup: " << (found_rev ? "PASS" : "FAIL") << "\n";
lp::lar_term t2;
t2.add_monomial(rational(1), x);
t2.add_monomial(rational(-1), y);
t2.add_monomial(rational(5), z);
s.add_term(t2.coeffs_as_vector(), UINT_MAX);
s.register_existing_terms();
lp::lar_term query2;
query2.add_monomial(rational(1), x);
query2.add_monomial(rational(-1), y);
query2.add_monomial(rational(5), z);
lp::lar_term norm2 = query2.get_normalized_by_min_var(a);
found = s.fetch_normalized_term_column(norm2, result);
VERIFY(found);
std::cout << " 3-variable term lookup: " << (found ? "PASS" : "FAIL") << "\n";
lp::lar_term query3;
query3.add_monomial(rational(7), x);
query3.add_monomial(rational(11), y);
lp::lar_term norm3 = query3.get_normalized_by_min_var(a);
bool found_missing = s.fetch_normalized_term_column(norm3, result);
VERIFY(!found_missing);
std::cout << " non-existent term not found: " << (!found_missing ? "PASS" : "FAIL") << "\n";
}
void test_nla_intervals() {
test_nla_intervals_basic();
test_nla_intervals_negative();
test_nla_intervals_zero_crossing();
test_nla_intervals_power();
test_nla_intervals_mixed_signs();
test_nla_intervals_fractional();
test_fetch_normalized_term_column();
}
}
void tst_nla_intervals() {
nla::test_nla_intervals();
}