mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-03 21:44:30 +08:00
305 lines
7.0 KiB
C++
Executable File
305 lines
7.0 KiB
C++
Executable File
// $Id: test.cpp 4957 2006-02-17 23:44:38Z glandrum $
|
|
//
|
|
// Copyright (c) 2003-2006 Greg Landrum and Rational Discovery LLC
|
|
//
|
|
// @@ All Rights Reserved @@
|
|
//
|
|
|
|
#include "QueryObjects.h"
|
|
#include <iostream>
|
|
#include <math.h>
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
using namespace std;
|
|
using namespace Queries;
|
|
|
|
|
|
void test1(){
|
|
cout << "Float" << endl;
|
|
EqualityQuery<double> q(1.0);
|
|
|
|
CHECK_INVARIANT(!q.Match(0.0),"");
|
|
CHECK_INVARIANT(q.Match(1.0),"");
|
|
CHECK_INVARIANT(!q.Match(1.001),"");
|
|
CHECK_INVARIANT(!q.Match(1.1),"");
|
|
CHECK_INVARIANT(!q.Match(-2),"");
|
|
|
|
|
|
cout << "With Tolerance" << endl;
|
|
q.setTol(0.002);
|
|
CHECK_INVARIANT(!q.Match(0.0),"");
|
|
CHECK_INVARIANT(q.Match(1.0),"");
|
|
CHECK_INVARIANT(q.Match(1.001),"");
|
|
CHECK_INVARIANT(!q.Match(1.1),"");
|
|
CHECK_INVARIANT(!q.Match(-2),"");
|
|
|
|
Query<double> *newQ;
|
|
newQ = q.copy();
|
|
CHECK_INVARIANT(!newQ->Match(0.0),"");
|
|
CHECK_INVARIANT(newQ->Match(1.0),"");
|
|
CHECK_INVARIANT(newQ->Match(1.001),"");
|
|
CHECK_INVARIANT(!newQ->Match(1.1),"");
|
|
CHECK_INVARIANT(!newQ->Match(-2),"");
|
|
}
|
|
|
|
void test2(){
|
|
cout << "Set" << endl;
|
|
SetQuery<int> q;
|
|
q.insert(1);
|
|
q.insert(3);
|
|
q.insert(5);
|
|
|
|
CHECK_INVARIANT(!q.Match(0),"");
|
|
CHECK_INVARIANT(q.Match(1),"");
|
|
CHECK_INVARIANT(q.Match(3),"");
|
|
CHECK_INVARIANT(!q.Match(-3),"");
|
|
|
|
Query<int> *newQ;
|
|
newQ = q.copy();
|
|
CHECK_INVARIANT(!newQ->Match(0),"");
|
|
CHECK_INVARIANT(newQ->Match(1),"");
|
|
CHECK_INVARIANT(newQ->Match(3),"");
|
|
CHECK_INVARIANT(!newQ->Match(-3),"");
|
|
|
|
}
|
|
|
|
void test3(){
|
|
cout << "And" << endl;
|
|
AndQuery<int> *q = new AndQuery<int>;
|
|
LessQuery<int> *l = new LessQuery<int>;
|
|
l->setVal(0);
|
|
GreaterEqualQuery<int> *g = new GreaterEqualQuery<int>;
|
|
g->setVal(3);
|
|
|
|
q->addChild(Query<int>::CHILD_TYPE(l));
|
|
q->addChild(Query<int>::CHILD_TYPE(g));
|
|
|
|
CHECK_INVARIANT(!q->Match(0),"");
|
|
CHECK_INVARIANT(q->Match(1),"");
|
|
CHECK_INVARIANT(q->Match(3),"");
|
|
CHECK_INVARIANT(!q->Match(-3),"");
|
|
|
|
Query<int> *newQ;
|
|
newQ = q->copy();
|
|
CHECK_INVARIANT(!newQ->Match(0),"");
|
|
CHECK_INVARIANT(newQ->Match(1),"");
|
|
CHECK_INVARIANT(newQ->Match(3),"");
|
|
CHECK_INVARIANT(!newQ->Match(-3),"");
|
|
|
|
}
|
|
|
|
void test4(){
|
|
cout << "Or" << endl;
|
|
OrQuery<int> *q = new OrQuery<int>;
|
|
LessQuery<int> *l = new LessQuery<int>;
|
|
l->setVal(0);
|
|
GreaterEqualQuery<int> *g = new GreaterEqualQuery<int>;
|
|
g->setVal(3);
|
|
|
|
q->addChild(Query<int>::CHILD_TYPE(l));
|
|
q->addChild(Query<int>::CHILD_TYPE(g));
|
|
|
|
CHECK_INVARIANT(q->Match(0),"");
|
|
CHECK_INVARIANT(q->Match(1),"");
|
|
CHECK_INVARIANT(q->Match(3),"");
|
|
CHECK_INVARIANT(q->Match(-3),"");
|
|
|
|
Query<int> *newQ;
|
|
newQ = q->copy();
|
|
CHECK_INVARIANT(newQ->Match(0),"");
|
|
CHECK_INVARIANT(newQ->Match(1),"");
|
|
CHECK_INVARIANT(newQ->Match(3),"");
|
|
CHECK_INVARIANT(newQ->Match(-3),"");
|
|
|
|
}
|
|
|
|
void test5(){
|
|
cout << "XOr" << endl;
|
|
XOrQuery<int> *q = new XOrQuery<int>;
|
|
LessQuery<int> *l = new LessQuery<int>;
|
|
l->setVal(0);
|
|
GreaterEqualQuery<int> *g = new GreaterEqualQuery<int>;
|
|
g->setVal(3);
|
|
|
|
q->addChild(Query<int>::CHILD_TYPE(l));
|
|
q->addChild(Query<int>::CHILD_TYPE(g));
|
|
|
|
CHECK_INVARIANT(q->Match(-1),"");
|
|
CHECK_INVARIANT(q->Match(0),"");
|
|
CHECK_INVARIANT(!q->Match(1),"");
|
|
CHECK_INVARIANT(!q->Match(3),"");
|
|
CHECK_INVARIANT(q->Match(-3),"");
|
|
|
|
Query<int> *newQ;
|
|
newQ = q->copy();
|
|
CHECK_INVARIANT(newQ->Match(-1),"");
|
|
CHECK_INVARIANT(newQ->Match(0),"");
|
|
CHECK_INVARIANT(!newQ->Match(1),"");
|
|
CHECK_INVARIANT(!newQ->Match(3),"");
|
|
CHECK_INVARIANT(newQ->Match(-3),"");
|
|
}
|
|
|
|
int foofun(double bar) { return int(floor(bar)); };
|
|
|
|
void test6(){
|
|
cout << "pointer and copy foo" << endl;
|
|
EqualityQuery<int,double,true> q;
|
|
q.setDataFunc(foofun);
|
|
q.setVal(6);
|
|
CHECK_INVARIANT(q.Match(6.0),"");
|
|
CHECK_INVARIANT(q.Match(6.1),"");
|
|
CHECK_INVARIANT(!q.Match(5.0),"");
|
|
|
|
Query<int,double,true> *newQ;
|
|
newQ = q.copy();
|
|
CHECK_INVARIANT(newQ->Match(6.0),"");
|
|
CHECK_INVARIANT(newQ->Match(6.1),"");
|
|
CHECK_INVARIANT(!newQ->Match(5.0),"");
|
|
|
|
Query<int,double,true> *newQ2 = &q;
|
|
CHECK_INVARIANT(newQ2->Match(6.0),"");
|
|
CHECK_INVARIANT(newQ2->Match(6.1),"");
|
|
CHECK_INVARIANT(!newQ2->Match(5.0),"");
|
|
|
|
Query<int,double,true> *newQ3;
|
|
newQ3 = newQ2->copy();
|
|
CHECK_INVARIANT(newQ3->Match(6.0),"");
|
|
CHECK_INVARIANT(newQ3->Match(6.1),"");
|
|
CHECK_INVARIANT(!newQ3->Match(5.0),"");
|
|
|
|
|
|
}
|
|
|
|
bool matchF(int v){
|
|
return v == 3;
|
|
}
|
|
|
|
int dataF(float v){
|
|
return int(floor(v))*3;
|
|
}
|
|
|
|
bool cmp(int v){
|
|
return v < 3;
|
|
}
|
|
|
|
void basics1(){
|
|
cout << "Query" << endl;
|
|
Query<int,float,true> q;
|
|
q.setMatchFunc(matchF);
|
|
q.setDataFunc(dataF);
|
|
|
|
CHECK_INVARIANT(!q.Match(0.0),"");
|
|
CHECK_INVARIANT(q.Match(1.0),"");
|
|
CHECK_INVARIANT(q.Match(1.1),"");
|
|
CHECK_INVARIANT(!q.Match(-2.0),"");
|
|
|
|
cout << "Query2" << endl;
|
|
Query<bool,int,true> q2;
|
|
q2.setDataFunc(cmp);
|
|
CHECK_INVARIANT(q2.Match(0),"");
|
|
CHECK_INVARIANT(q2.Match(1),"");
|
|
CHECK_INVARIANT(!q2.Match(3),"");
|
|
CHECK_INVARIANT(!q2.Match(4),"");
|
|
CHECK_INVARIANT(!q2.Match(4.0),"");
|
|
}
|
|
|
|
void basics2(){
|
|
cout << "Equality" << endl;
|
|
EqualityQuery<int> q2;
|
|
q2.setVal(3);
|
|
CHECK_INVARIANT(!q2.Match(0),"");
|
|
CHECK_INVARIANT(!q2.Match(1),"");
|
|
CHECK_INVARIANT(q2.Match(3),"");
|
|
CHECK_INVARIANT(!q2.Match(-3),"");
|
|
|
|
cout << "Greater" << endl;
|
|
GreaterQuery<int> q3;
|
|
q3.setVal(3);
|
|
CHECK_INVARIANT(q3.Match(0),"");
|
|
CHECK_INVARIANT(q3.Match(1),"");
|
|
CHECK_INVARIANT(!q3.Match(3),"");
|
|
CHECK_INVARIANT(!q3.Match(5),"");
|
|
|
|
|
|
cout << "GreaterEqual" << endl;
|
|
GreaterEqualQuery<int> q4(3);
|
|
CHECK_INVARIANT(q4.Match(0),"");
|
|
CHECK_INVARIANT(q4.Match(1),"");
|
|
CHECK_INVARIANT(q4.Match(3),"");
|
|
CHECK_INVARIANT(!q4.Match(5),"");
|
|
|
|
cout << "Less" << endl;
|
|
LessQuery<int> q5;
|
|
q5.setVal(3);
|
|
CHECK_INVARIANT(!q5.Match(0),"");
|
|
CHECK_INVARIANT(!q5.Match(1),"");
|
|
CHECK_INVARIANT(!q5.Match(3),"");
|
|
CHECK_INVARIANT(q5.Match(5),"");
|
|
|
|
cout << "LessEqual" << endl;
|
|
LessEqualQuery<int> q6(3);
|
|
|
|
CHECK_INVARIANT(!q6.Match(0),"");
|
|
CHECK_INVARIANT(!q6.Match(1),"");
|
|
CHECK_INVARIANT(q6.Match(3),"");
|
|
CHECK_INVARIANT(q6.Match(5),"");
|
|
|
|
|
|
cout << "Open Range" << endl;
|
|
RangeQuery<int> q7(0,3);
|
|
CHECK_INVARIANT(!q7.Match(0),"");
|
|
CHECK_INVARIANT(q7.Match(1),"");
|
|
CHECK_INVARIANT(!q7.Match(3),"");
|
|
CHECK_INVARIANT(!q7.Match(5),"");
|
|
|
|
|
|
cout << "Closed Range" << endl;
|
|
q7.setEndsOpen(false,false);
|
|
CHECK_INVARIANT(q7.Match(0),"");
|
|
CHECK_INVARIANT(q7.Match(1),"");
|
|
CHECK_INVARIANT(q7.Match(3),"");
|
|
CHECK_INVARIANT(!q7.Match(5),"");
|
|
|
|
}
|
|
|
|
|
|
int convFunc(const char*arg) {return boost::lexical_cast<int>(arg);};
|
|
|
|
void test7(){
|
|
cout << "Set2" << endl;
|
|
SetQuery<int,const char*,true> q;
|
|
q.setDataFunc(convFunc);
|
|
q.insert(1);
|
|
q.insert(3);
|
|
q.insert(5);
|
|
|
|
CHECK_INVARIANT(!q.Match("0"),"");
|
|
CHECK_INVARIANT(q.Match("1"),"");
|
|
CHECK_INVARIANT(q.Match("3"),"");
|
|
CHECK_INVARIANT(!q.Match("-3"),"");
|
|
|
|
Query<int,const char*,true> *newQ;
|
|
newQ = q.copy();
|
|
CHECK_INVARIANT(!newQ->Match("0"),"");
|
|
CHECK_INVARIANT(newQ->Match("1"),"");
|
|
CHECK_INVARIANT(newQ->Match("3"),"");
|
|
CHECK_INVARIANT(!newQ->Match("-3"),"");
|
|
|
|
}
|
|
|
|
int main()
|
|
{
|
|
basics1();
|
|
basics2();
|
|
|
|
|
|
test1();
|
|
test2();
|
|
test3();
|
|
test4();
|
|
test5();
|
|
test6();
|
|
test7();
|
|
return 0;
|
|
}
|