1 | /*
|
---|
2 | * Project: MoleCuilder
|
---|
3 | * Description: creates and alters molecular systems
|
---|
4 | * Copyright (C) 2010 University of Bonn. All rights reserved.
|
---|
5 | * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
|
---|
6 | */
|
---|
7 |
|
---|
8 | /*
|
---|
9 | * atomsCalculationTest.cpp
|
---|
10 | *
|
---|
11 | * Created on: Feb 19, 2010
|
---|
12 | * Author: crueger
|
---|
13 | */
|
---|
14 |
|
---|
15 | // include config.h
|
---|
16 | #ifdef HAVE_CONFIG_H
|
---|
17 | #include <config.h>
|
---|
18 | #endif
|
---|
19 |
|
---|
20 | #include "atomsCalculationTest.hpp"
|
---|
21 |
|
---|
22 | #include <cppunit/CompilerOutputter.h>
|
---|
23 | #include <cppunit/extensions/TestFactoryRegistry.h>
|
---|
24 | #include <cppunit/ui/text/TestRunner.h>
|
---|
25 | #include <iostream>
|
---|
26 | #include <boost/bind.hpp>
|
---|
27 |
|
---|
28 | #include "Descriptors/AtomDescriptor.hpp"
|
---|
29 | #include "Descriptors/AtomIdDescriptor.hpp"
|
---|
30 | #include "Actions/AtomsCalculation.hpp"
|
---|
31 | #include "Actions/AtomsCalculation_impl.hpp"
|
---|
32 | #include "Actions/ActionRegistry.hpp"
|
---|
33 | #include "Actions/ActionTraits.hpp"
|
---|
34 |
|
---|
35 | #include "World.hpp"
|
---|
36 | #include "World_calculations.hpp"
|
---|
37 | #include "atom.hpp"
|
---|
38 |
|
---|
39 | #ifdef HAVE_TESTRUNNER
|
---|
40 | #include "UnitTestMain.hpp"
|
---|
41 | #endif /*HAVE_TESTRUNNER*/
|
---|
42 |
|
---|
43 | // Registers the fixture into the 'registry'
|
---|
44 | CPPUNIT_TEST_SUITE_REGISTRATION( atomsCalculationTest );
|
---|
45 |
|
---|
46 | // set up and tear down
|
---|
47 | void atomsCalculationTest::setUp(){
|
---|
48 | World::getInstance();
|
---|
49 | for(int i=0;i<ATOM_COUNT;++i){
|
---|
50 | atoms[i]= World::getInstance().createAtom();
|
---|
51 | atomIds[i]= atoms[i]->getId();
|
---|
52 | }
|
---|
53 | }
|
---|
54 | void atomsCalculationTest::tearDown(){
|
---|
55 | World::purgeInstance();
|
---|
56 | ActionRegistry::purgeInstance();
|
---|
57 | }
|
---|
58 |
|
---|
59 | // some helper functions
|
---|
60 | static bool hasAllIds(std::vector<atomId_t> atoms,atomId_t ids[ATOM_COUNT], std::set<atomId_t> excluded = std::set<atomId_t>()){
|
---|
61 | for(int i=0;i<ATOM_COUNT;++i){
|
---|
62 | atomId_t id = ids[i];
|
---|
63 | if(!excluded.count(id)){
|
---|
64 | std::vector<atomId_t>::iterator iter;
|
---|
65 | bool res=false;
|
---|
66 | for(iter=atoms.begin();iter!=atoms.end();++iter){
|
---|
67 | res |= (*iter) == id;
|
---|
68 | }
|
---|
69 | if(!res) {
|
---|
70 | cout << "Atom " << id << " missing in returned list" << endl;
|
---|
71 | return false;
|
---|
72 | }
|
---|
73 | }
|
---|
74 | }
|
---|
75 | return true;
|
---|
76 | }
|
---|
77 |
|
---|
78 | static bool hasNoDuplicates(std::vector<atomId_t> ids){
|
---|
79 | std::set<atomId_t> found;
|
---|
80 | std::vector<atomId_t>::iterator iter;
|
---|
81 | for(iter=ids.begin();iter!=ids.end();++iter){
|
---|
82 | int id = (*iter);
|
---|
83 | if(found.count(id))
|
---|
84 | return false;
|
---|
85 | found.insert(id);
|
---|
86 | }
|
---|
87 | return true;
|
---|
88 | }
|
---|
89 |
|
---|
90 | void atomsCalculationTest::testCalculateSimple(){
|
---|
91 | ActionTraits FooTrait("FOO");
|
---|
92 | AtomsCalculation<atomId_t> *calc = World::getInstance().calcOnAtoms<atomId_t>(boost::bind(&atom::getId,_1),FooTrait,AllAtoms());
|
---|
93 | std::vector<atomId_t> allIds = (*calc)();
|
---|
94 | CPPUNIT_ASSERT(hasAllIds(allIds,atomIds));
|
---|
95 | CPPUNIT_ASSERT(hasNoDuplicates(allIds));
|
---|
96 | }
|
---|
97 |
|
---|
98 | void atomsCalculationTest::testCalculateExcluded(){
|
---|
99 | ActionTraits FooTrait("FOO");
|
---|
100 | atomId_t excluded = atomIds[ATOM_COUNT/2];
|
---|
101 | AtomsCalculation<atomId_t> *calc = World::getInstance().calcOnAtoms<atomId_t>(boost::bind(&atom::getId,_1),FooTrait,AllAtoms() && !AtomById(excluded));
|
---|
102 | std::vector<atomId_t> allIds = (*calc)();
|
---|
103 | std::set<atomId_t> excluded_set;
|
---|
104 | excluded_set.insert(excluded);
|
---|
105 | CPPUNIT_ASSERT(hasAllIds(allIds,atomIds,excluded_set));
|
---|
106 | CPPUNIT_ASSERT(hasNoDuplicates(allIds));
|
---|
107 | CPPUNIT_ASSERT_EQUAL((size_t)(ATOM_COUNT-1),allIds.size());
|
---|
108 | }
|
---|