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 | * CountBondsUnitTest.cpp
|
---|
10 | *
|
---|
11 | * Created on: Mar 30, 2010
|
---|
12 | * Author: heber
|
---|
13 | */
|
---|
14 |
|
---|
15 | // include config.h
|
---|
16 | #ifdef HAVE_CONFIG_H
|
---|
17 | #include <config.h>
|
---|
18 | #endif
|
---|
19 |
|
---|
20 | using namespace std;
|
---|
21 |
|
---|
22 | #include <cppunit/CompilerOutputter.h>
|
---|
23 | #include <cppunit/extensions/TestFactoryRegistry.h>
|
---|
24 | #include <cppunit/ui/text/TestRunner.h>
|
---|
25 |
|
---|
26 | #include <iostream>
|
---|
27 | #include <stdio.h>
|
---|
28 | #include <cstring>
|
---|
29 |
|
---|
30 | #include "CodePatterns/Assert.hpp"
|
---|
31 |
|
---|
32 | #include "Analysis/analysis_bonds.hpp"
|
---|
33 | #include "atom.hpp"
|
---|
34 | #include "Bond/bond.hpp"
|
---|
35 | #include "Element/element.hpp"
|
---|
36 | #include "Graph/BondGraph.hpp"
|
---|
37 | #include "molecule.hpp"
|
---|
38 | #include "Element/periodentafel.hpp"
|
---|
39 | #include "World.hpp"
|
---|
40 |
|
---|
41 | #include "CountBondsUnitTest.hpp"
|
---|
42 |
|
---|
43 | #ifdef HAVE_TESTRUNNER
|
---|
44 | #include "UnitTestMain.hpp"
|
---|
45 | #endif /*HAVE_TESTRUNNER*/
|
---|
46 |
|
---|
47 | /********************************************** Test classes **************************************/
|
---|
48 |
|
---|
49 | // Registers the fixture into the 'registry'
|
---|
50 | CPPUNIT_TEST_SUITE_REGISTRATION( CountBondsTest );
|
---|
51 |
|
---|
52 |
|
---|
53 | void CountBondsTest::setUp()
|
---|
54 | {
|
---|
55 | atom *Walker = NULL;
|
---|
56 |
|
---|
57 | // construct element
|
---|
58 | hydrogen = World::getInstance().getPeriode()->FindElement(1);
|
---|
59 | oxygen = World::getInstance().getPeriode()->FindElement(8);
|
---|
60 | CPPUNIT_ASSERT(hydrogen != NULL && "could not find element hydrogen");
|
---|
61 | CPPUNIT_ASSERT(oxygen != NULL && "could not find element oxygen");
|
---|
62 |
|
---|
63 | // construct molecule (water molecule)
|
---|
64 | TestMolecule1 = World::getInstance().createMolecule();
|
---|
65 | CPPUNIT_ASSERT(TestMolecule1 != NULL && "could not create first molecule");
|
---|
66 | Walker = World::getInstance().createAtom();
|
---|
67 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
68 | Walker->setType(hydrogen);
|
---|
69 | Walker->setPosition(Vector(-0.2418, 0.9350, 0. ));
|
---|
70 | TestMolecule1->AddAtom(Walker);
|
---|
71 | Walker = World::getInstance().createAtom();
|
---|
72 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
73 | Walker->setType(hydrogen);
|
---|
74 | Walker->setPosition(Vector(0.9658, 0., 0. ));
|
---|
75 | TestMolecule1->AddAtom(Walker);
|
---|
76 | Walker = World::getInstance().createAtom();
|
---|
77 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
78 | Walker->setType(oxygen);
|
---|
79 | Walker->setPosition(Vector(0., 0., 0. ));
|
---|
80 | TestMolecule1->AddAtom(Walker);
|
---|
81 |
|
---|
82 | TestMolecule2 = World::getInstance().createMolecule();
|
---|
83 | CPPUNIT_ASSERT(TestMolecule2 != NULL && "could not create second molecule");
|
---|
84 | Walker = World::getInstance().createAtom();
|
---|
85 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
86 | Walker->setType(hydrogen);
|
---|
87 | Walker->setPosition(Vector(-0.2418, 0.9350, 0. ));
|
---|
88 | TestMolecule2->AddAtom(Walker);
|
---|
89 | Walker = World::getInstance().createAtom();
|
---|
90 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
91 | Walker->setType(hydrogen);
|
---|
92 | Walker->setPosition(Vector(0.9658, 0., 0. ));
|
---|
93 | TestMolecule2->AddAtom(Walker);
|
---|
94 | Walker = World::getInstance().createAtom();
|
---|
95 | CPPUNIT_ASSERT(Walker != NULL && "could not create atom");
|
---|
96 | Walker->setType(oxygen);
|
---|
97 | Walker->setPosition(Vector(0., 0., 0. ));
|
---|
98 | TestMolecule2->AddAtom(Walker);
|
---|
99 |
|
---|
100 | molecules = World::getInstance().getMolecules();
|
---|
101 | CPPUNIT_ASSERT(molecules != NULL && "could not obtain list of molecules");
|
---|
102 | molecules->insert(TestMolecule1);
|
---|
103 | molecules->insert(TestMolecule2);
|
---|
104 |
|
---|
105 | // check that TestMolecule was correctly constructed
|
---|
106 | CPPUNIT_ASSERT_EQUAL( TestMolecule1->getAtomCount(), 3 );
|
---|
107 | CPPUNIT_ASSERT_EQUAL( TestMolecule2->getAtomCount(), 3 );
|
---|
108 |
|
---|
109 | // create a small file with table
|
---|
110 | BG = new BondGraph(true);
|
---|
111 | CPPUNIT_ASSERT(BG != NULL && "could not create BondGraph");
|
---|
112 |
|
---|
113 | // construct bond graphs
|
---|
114 | World::AtomComposite Set1 = TestMolecule1->getAtomSet();
|
---|
115 | BG->CreateAdjacency(Set1);
|
---|
116 | CPPUNIT_ASSERT( TestMolecule1->getBondCount() != 0);
|
---|
117 | World::AtomComposite Set2 = TestMolecule2->getAtomSet();
|
---|
118 | BG->CreateAdjacency(Set2);
|
---|
119 | CPPUNIT_ASSERT( TestMolecule2->getBondCount() != 0);
|
---|
120 | // TestMolecule1->Output((ofstream *)&cout);
|
---|
121 | // TestMolecule1->OutputBondsList(std::cout);
|
---|
122 | };
|
---|
123 |
|
---|
124 |
|
---|
125 | void CountBondsTest::tearDown()
|
---|
126 | {
|
---|
127 | // remove the file
|
---|
128 | delete(BG);
|
---|
129 |
|
---|
130 | World::purgeInstance();
|
---|
131 | };
|
---|
132 |
|
---|
133 | /** UnitTest for CountBondsTest::BondsOfTwoTest().
|
---|
134 | */
|
---|
135 | void CountBondsTest::BondsOfTwoTest()
|
---|
136 | {
|
---|
137 | CPPUNIT_ASSERT_EQUAL( 4 , CountBondsOfTwo(molecules, hydrogen, oxygen) );
|
---|
138 | CPPUNIT_ASSERT_EQUAL( 0 , CountBondsOfTwo(molecules, hydrogen, hydrogen) );
|
---|
139 | CPPUNIT_ASSERT_EQUAL( 0 , CountBondsOfTwo(molecules, oxygen, oxygen) );
|
---|
140 | };
|
---|
141 |
|
---|
142 | /** UnitTest for CountBondsTest::BondsOfThreeTest().
|
---|
143 | */
|
---|
144 | void CountBondsTest::BondsOfThreeTest()
|
---|
145 | {
|
---|
146 | CPPUNIT_ASSERT_EQUAL( 2 , CountBondsOfThree(molecules, hydrogen, oxygen, hydrogen) );
|
---|
147 | CPPUNIT_ASSERT_EQUAL( 0 , CountBondsOfThree(molecules, oxygen, hydrogen, oxygen) );
|
---|
148 | };
|
---|
149 |
|
---|
150 | void OutputTestMolecule(molecule *mol, const char *name)
|
---|
151 | {
|
---|
152 | ofstream output(name);
|
---|
153 | mol->OutputXYZ(&output);
|
---|
154 | output.close();
|
---|
155 | }
|
---|
156 |
|
---|
157 | /** UnitTest for CountBondsTest::HydrogenBridgeBondsTest().
|
---|
158 | */
|
---|
159 | void CountBondsTest::HydrogenBridgeBondsTest()
|
---|
160 | {
|
---|
161 | double *mirror = new double[3];
|
---|
162 | CPPUNIT_ASSERT(mirror != NULL && "could not create array of doubles");
|
---|
163 | for (int i=0;i<3;i++)
|
---|
164 | mirror[i] = -1.;
|
---|
165 | Vector Translator;
|
---|
166 |
|
---|
167 | //OutputTestMolecule(TestMolecule1, "testmolecule1.xyz");
|
---|
168 |
|
---|
169 | cout << "Case 1: offset of (3,0,0), hence angles are (104.5, 0, 75.5, 180) < 30." << endl;
|
---|
170 | Translator = Vector(3,0,0);
|
---|
171 | TestMolecule2->Translate(&Translator);
|
---|
172 | CPPUNIT_ASSERT_EQUAL( 1 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
173 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, oxygen, NULL) );
|
---|
174 | //OutputTestMolecule(TestMolecule2, "testmolecule2-1.xyz");
|
---|
175 | Translator = Vector(-3,0,0);
|
---|
176 | TestMolecule2->Translate(&Translator);
|
---|
177 |
|
---|
178 | cout << "Case 2: offset of (0,3,0), hence angle are (14.5, 165.5, 90) < 30 (only three, because other 90 is missing due to first H01 only fulfilling H-bond criteria)." << endl;
|
---|
179 | Translator = Vector(0,3,0);
|
---|
180 | TestMolecule2->Translate(&Translator);
|
---|
181 | CPPUNIT_ASSERT_EQUAL( 1 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
182 | //OutputTestMolecule(TestMolecule2, "testmolecule2-2.xyz");
|
---|
183 | Translator = Vector(0,-3,0);
|
---|
184 | TestMolecule2->Translate(&Translator);
|
---|
185 |
|
---|
186 | cout << "Case 3: offset of (0,-3,0) and mirror, hence angle are (165.5, 90, 165.5, 90) > 30." << endl;
|
---|
187 | Translator = Vector(0,-3,0);
|
---|
188 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
189 | TestMolecule2->Translate(&Translator);
|
---|
190 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
191 | //OutputTestMolecule(TestMolecule2, "testmolecule2-3.xyz");
|
---|
192 | Translator = Vector(0,3,0);
|
---|
193 | TestMolecule2->Translate(&Translator);
|
---|
194 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
195 |
|
---|
196 | cout << "Case 4: offset of (2,1,0), hence angle are (78, 26.6, 102, 153.4) < 30." << endl;
|
---|
197 | Translator = Vector(2,1,0);
|
---|
198 | TestMolecule2->Translate(&Translator);
|
---|
199 | CPPUNIT_ASSERT_EQUAL( 1 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
200 | //OutputTestMolecule(TestMolecule2, "testmolecule2-4.xyz");
|
---|
201 | Translator = Vector(-2,-1,0);
|
---|
202 | TestMolecule2->Translate(&Translator);
|
---|
203 |
|
---|
204 | cout << "Case 5: offset of (0,0,3), hence angle are (90, 90, 90, 90) > 30." << endl;
|
---|
205 | Translator = Vector(0,0,3);
|
---|
206 | TestMolecule2->Translate(&Translator);
|
---|
207 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
208 | //OutputTestMolecule(TestMolecule2, "testmolecule2-5.xyz");
|
---|
209 | Translator = Vector(0,0,-3);
|
---|
210 | TestMolecule2->Translate(&Translator);
|
---|
211 |
|
---|
212 | cout << "Case 6: offset of (-3,0,0) and mirror, hence angle are (75.5, 180, 104.5, 180) > 30." << endl;
|
---|
213 | Translator = Vector(-3,0,0);
|
---|
214 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
215 | TestMolecule2->Translate(&Translator);
|
---|
216 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
217 | //OutputTestMolecule(TestMolecule2, "testmolecule2-6.xyz");
|
---|
218 | Translator = Vector(3,0,0);
|
---|
219 | TestMolecule2->Translate(&Translator);
|
---|
220 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
221 |
|
---|
222 | cout << "Case 7: offset of (3,0,0) and mirror, hence angles are (104.5, 0, 104.5, 0) < 30, but interfering hydrogens." << endl;
|
---|
223 | Translator = Vector(3,0,0);
|
---|
224 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
225 | TestMolecule2->Translate(&Translator);
|
---|
226 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
227 | //OutputTestMolecule(TestMolecule2, "testmolecule2-7.xyz");
|
---|
228 | Translator = Vector(-3,0,0);
|
---|
229 | TestMolecule2->Translate(&Translator);
|
---|
230 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
231 |
|
---|
232 | cout << "Case 8: offset of (0,3,0), hence angle are (14.5, 90, 14.5, 90) < 30, but interfering hydrogens." << endl;
|
---|
233 | Translator = Vector(0,3,0);
|
---|
234 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
235 | TestMolecule2->Translate(&Translator);
|
---|
236 | //OutputTestMolecule(TestMolecule2, "testmolecule2-8.xyz");
|
---|
237 | CPPUNIT_ASSERT_EQUAL( 0 , CountHydrogenBridgeBonds(molecules, NULL, NULL) );
|
---|
238 | Translator = Vector(0,-3,0);
|
---|
239 | TestMolecule2->Translate(&Translator);
|
---|
240 | TestMolecule2->Scale((const double ** const)&mirror);
|
---|
241 |
|
---|
242 | delete[](mirror);
|
---|
243 | };
|
---|