mirror of
https://github.com/Z3Prover/z3
synced 2025-04-23 17:15:31 +00:00
move doc code to rel, adding unit test
Signed-off-by: Nikolaj Bjorner <nbjorner@microsoft.com>
This commit is contained in:
Nikolaj Bjorner
parent
cb8ad76677
commit
5679cc7567
12 changed files with 1138 additions and 19 deletions
|
|
@ -224,7 +224,7 @@ class test_doc_cls {
|
|||
|
||||
void test_merge(unsigned num_clauses) {
|
||||
doc_ref d(dm, dm.allocateX());
|
||||
expr_ref_vector fmls(m);
|
||||
expr_ref_vector fmls(m), eqs(m);
|
||||
th_rewriter rw(m);
|
||||
unsigned N = m_vars.size();
|
||||
expr_ref fml1(m), fml2(m), fml3(m), tmp1(m), tmp2(m), fml(m);
|
||||
|
|
@ -251,15 +251,26 @@ class test_doc_cls {
|
|||
for (unsigned i = 0; i < N; ++i) {
|
||||
if (to_merge[i] && i != lo) {
|
||||
equalities.merge(i, lo);
|
||||
eqs.push_back(m.mk_eq(m_vars[i].get(), m_vars[lo].get()));
|
||||
}
|
||||
}
|
||||
fml1 = to_formula(*d, dm, to_delete.c_ptr());
|
||||
eqs.push_back(to_formula(*d, dm, to_delete.c_ptr()));
|
||||
fml1 = mk_and(m, eqs.size(), eqs.c_ptr());
|
||||
if (dm.merge(*d, lo, 1, equalities, discard_cols)) {
|
||||
fml2 = to_formula(*d, dm, to_delete.c_ptr());
|
||||
std::cout << fml1 << "\n";
|
||||
std::cout << fml2 << "\n";
|
||||
}
|
||||
// ...
|
||||
else {
|
||||
fml2 = m.mk_false();
|
||||
}
|
||||
rw(fml1);
|
||||
rw(fml2);
|
||||
smt_params fp;
|
||||
smt::kernel solver(m, fp);
|
||||
TRACE("doc", tout << "manual project:\n" << fml1 << "\nautomatic project: \n" << fml2 << "\n";);
|
||||
fml = m.mk_not(m.mk_eq(fml1, fml2));
|
||||
solver.assert_expr(fml);
|
||||
lbool res = solver.check();
|
||||
SASSERT(res == l_false);
|
||||
}
|
||||
|
||||
public:
|
||||
|
|
|
|||
|
|
@ -143,6 +143,7 @@ int main(int argc, char ** argv) {
|
|||
TST(fixed_bit_vector);
|
||||
TST(tbv);
|
||||
TST(doc);
|
||||
TST(udoc_relation);
|
||||
TST(string_buffer);
|
||||
TST(map);
|
||||
TST(diff_logic);
|
||||
|
|
|
|||
211
src/test/udoc_relation.cpp
Normal file
211
src/test/udoc_relation.cpp
Normal file
|
|
@ -0,0 +1,211 @@
|
|||
#include "udoc_relation.h"
|
||||
#include "trace.h"
|
||||
#include "vector.h"
|
||||
#include "ast.h"
|
||||
#include "ast_pp.h"
|
||||
#include "reg_decl_plugins.h"
|
||||
#include "sorting_network.h"
|
||||
#include "smt_kernel.h"
|
||||
#include "model_smt2_pp.h"
|
||||
#include "smt_params.h"
|
||||
#include "ast_util.h"
|
||||
#include "expr_safe_replace.h"
|
||||
#include "th_rewriter.h"
|
||||
#include "dl_relation_manager.h"
|
||||
#include "dl_register_engine.h"
|
||||
#include "rel_context.h"
|
||||
#include "bv_decl_plugin.h"
|
||||
|
||||
class udoc_tester {
|
||||
typedef datalog::relation_base relation_base;
|
||||
typedef datalog::udoc_relation udoc_relation;
|
||||
typedef datalog::udoc_plugin udoc_plugin;
|
||||
typedef datalog::relation_signature relation_signature;
|
||||
|
||||
struct init {
|
||||
init(ast_manager& m) {
|
||||
reg_decl_plugins(m);
|
||||
}
|
||||
};
|
||||
random_gen m_rand;
|
||||
ast_manager m;
|
||||
init m_init;
|
||||
bv_util bv;
|
||||
expr_ref_vector m_vars;
|
||||
smt_params m_smt_params;
|
||||
datalog::register_engine m_reg;
|
||||
datalog::context m_ctx;
|
||||
datalog::rel_context rc;
|
||||
udoc_plugin& p;
|
||||
public:
|
||||
udoc_tester(): m_init(m), bv(m), m_vars(m), m_ctx(m, m_reg, m_smt_params), rc(m_ctx),
|
||||
p(dynamic_cast<udoc_plugin&>(*rc.get_rmanager().get_relation_plugin(symbol("doc"))))
|
||||
{
|
||||
}
|
||||
|
||||
udoc_relation* mk_empty(relation_signature const& sig) {
|
||||
SASSERT(p.can_handle_signature(sig));
|
||||
relation_base* empty = p.mk_empty(sig);
|
||||
return dynamic_cast<udoc_relation*>(empty);
|
||||
}
|
||||
|
||||
udoc_relation* mk_full(relation_signature const& sig) {
|
||||
func_decl_ref fn(m);
|
||||
fn = m.mk_func_decl(symbol("full"), sig.size(), sig.c_ptr(), m.mk_bool_sort());
|
||||
relation_base* full = p.mk_full(fn, sig);
|
||||
return dynamic_cast<udoc_relation*>(full);
|
||||
}
|
||||
|
||||
void test1() {
|
||||
datalog::relation_signature sig;
|
||||
sig.push_back(bv.mk_sort(12));
|
||||
sig.push_back(bv.mk_sort(6));
|
||||
sig.push_back(bv.mk_sort(12));
|
||||
|
||||
datalog::relation_fact fact1(m), fact2(m), fact3(m);
|
||||
fact1.push_back(bv.mk_numeral(rational(1), 12));
|
||||
fact1.push_back(bv.mk_numeral(rational(6), 6));
|
||||
fact1.push_back(bv.mk_numeral(rational(56), 12));
|
||||
fact2.push_back(bv.mk_numeral(rational(8), 12));
|
||||
fact2.push_back(bv.mk_numeral(rational(16), 6));
|
||||
fact2.push_back(bv.mk_numeral(rational(32), 12));
|
||||
fact3.push_back(bv.mk_numeral(rational(32), 12));
|
||||
fact3.push_back(bv.mk_numeral(rational(16), 6));
|
||||
fact3.push_back(bv.mk_numeral(rational(4), 12));
|
||||
|
||||
relation_base* t;
|
||||
expr_ref fml(m);
|
||||
// empty
|
||||
{
|
||||
std::cout << "empty\n";
|
||||
t = mk_empty(sig);
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->to_formula(fml);
|
||||
std::cout << fml << "\n";
|
||||
t->deallocate();
|
||||
}
|
||||
|
||||
// full
|
||||
{
|
||||
std::cout << "full\n";
|
||||
t = mk_full(sig);
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->to_formula(fml);
|
||||
std::cout << fml << "\n";
|
||||
t->deallocate();
|
||||
}
|
||||
|
||||
// join
|
||||
{
|
||||
udoc_relation* t1 = mk_full(sig);
|
||||
udoc_relation* t2 = mk_full(sig);
|
||||
udoc_relation* t3 = mk_empty(sig);
|
||||
unsigned_vector jc1, jc2;
|
||||
jc1.push_back(1);
|
||||
jc2.push_back(1);
|
||||
datalog::relation_join_fn* join_fn = p.mk_join_fn(*t1, *t2, jc1.size(), jc1.c_ptr(), jc2.c_ptr());
|
||||
SASSERT(join_fn);
|
||||
t = (*join_fn)(*t1, *t2);
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
t = (*join_fn)(*t1, *t3);
|
||||
SASSERT(t->empty());
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
t = (*join_fn)(*t3, *t3);
|
||||
SASSERT(t->empty());
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
dealloc(join_fn);
|
||||
t1->deallocate();
|
||||
t2->deallocate();
|
||||
t3->deallocate();
|
||||
}
|
||||
|
||||
// project
|
||||
{
|
||||
std::cout << "project\n";
|
||||
udoc_relation* t1 = mk_full(sig);
|
||||
unsigned_vector pc;
|
||||
pc.push_back(0);
|
||||
datalog::relation_transformer_fn* proj_fn = p.mk_project_fn(*t1, pc.size(), pc.c_ptr());
|
||||
t = (*proj_fn)(*t1);
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
t1->reset();
|
||||
t = (*proj_fn)(*t1);
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
t1->add_fact(fact1);
|
||||
t1->add_fact(fact2);
|
||||
t1->add_fact(fact3);
|
||||
t = (*proj_fn)(*t1);
|
||||
t1->display(std::cout); std::cout << "\n";
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
dealloc(proj_fn);
|
||||
t1->deallocate();
|
||||
}
|
||||
|
||||
// rename
|
||||
{
|
||||
udoc_relation* t1 = mk_empty(sig);
|
||||
unsigned_vector cycle;
|
||||
cycle.push_back(0);
|
||||
cycle.push_back(2);
|
||||
datalog::relation_transformer_fn* rename = p.mk_rename_fn(*t1, cycle.size(), cycle.c_ptr());
|
||||
|
||||
t1->add_fact(fact1);
|
||||
t1->add_fact(fact2);
|
||||
t1->add_fact(fact3);
|
||||
t = (*rename)(*t1);
|
||||
t1->display(std::cout); std::cout << "\n";
|
||||
t->display(std::cout); std::cout << "\n";
|
||||
t->deallocate();
|
||||
|
||||
dealloc(rename);
|
||||
t1->deallocate();
|
||||
}
|
||||
|
||||
// union
|
||||
{
|
||||
udoc_relation* t1 = mk_empty(sig);
|
||||
udoc_relation* t2 = mk_empty(sig);
|
||||
udoc_relation* delta = mk_full(sig);
|
||||
t2->add_fact(fact1);
|
||||
t2->add_fact(fact2);
|
||||
t1->add_fact(fact3);
|
||||
|
||||
datalog::relation_union_fn* union_fn = p.mk_union_fn(*t1, *t2, 0);
|
||||
|
||||
t1->display(std::cout << "t1 before:"); std::cout << "\n";
|
||||
(*union_fn)(*t1, *t2, delta);
|
||||
t1->display(std::cout << "t1 after:"); std::cout << "\n";
|
||||
delta->display(std::cout << "delta:"); std::cout << "\n";
|
||||
|
||||
t1->deallocate();
|
||||
t2->deallocate();
|
||||
delta->deallocate();
|
||||
}
|
||||
|
||||
// filter_identical
|
||||
|
||||
// filter_interpreted
|
||||
|
||||
// filter_by_negation
|
||||
|
||||
// filter_interpreted_project
|
||||
}
|
||||
};
|
||||
|
||||
void tst_udoc_relation() {
|
||||
udoc_tester tester;
|
||||
|
||||
tester.test1();
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue