File: BuildBondsProcessor_test.C

package info (click to toggle)
ball 1.5.0%2Bgit20180813.37fc53c-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 239,848 kB
  • sloc: cpp: 326,149; ansic: 4,208; python: 2,303; yacc: 1,778; lex: 1,099; xml: 958; sh: 322; makefile: 93
file content (117 lines) | stat: -rw-r--r-- 3,208 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//

#include <BALL/CONCEPT/classTest.h>
#include <BALLTestConfig.h>

///////////////////////////
#include <BALL/STRUCTURE/buildBondsProcessor.h>
#include <BALL/FORMAT/PDBFile.h>
#include <BALL/FORMAT/SDFile.h>
#include <BALL/KERNEL/forEach.h>
///////////////////////////

START_TEST(Fragment)

/////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////
using namespace BALL;
using namespace std;

BuildBondsProcessor* bbp = 0;
CHECK(BuildBondsProcessor())
	bbp = new BuildBondsProcessor();
	TEST_NOT_EQUAL(bbp, 0);
RESULT

CHECK(~BuildBondsProcessor())
	delete bbp;
RESULT

CHECK(BuildBondsProcessor(const String& filename))
	bbp = 0;
	bbp = new BuildBondsProcessor("bond_lengths/bond_lengths.db");
	TEST_NOT_EQUAL(bbp,0)
	TEST_EXCEPTION(Exception::FileNotFound, BuildBondsProcessor("file_does_not_exist"))
	delete bbp;
RESULT

CHECK(operator() (AtomContainer& ac))

	BuildBondsProcessor bbp2;
	bbp2.options[BuildBondsProcessor::Option::DELETE_EXISTING_BONDS] = "false";
	bbp2.options[BuildBondsProcessor::Option::REESTIMATE_BONDORDERS_RINGS] = "false";
	bbp2.options[BuildBondsProcessor::Option::DELETE_OVERESTIMATED_BONDS] = "false";
	
	PDBFile infileA(BALL_TEST_DATA_PATH(ACE_test_A.pdb));
	System sysA;
	infileA >> sysA;
	sysA.apply(bbp2);
	TEST_EQUAL(sysA.countBonds(), 1666)

	PDBFile infileB(BALL_TEST_DATA_PATH(ACE_test_B.pdb));
	System sysB;
	infileB >> sysB;
	sysB.apply(bbp2);
	TEST_EQUAL(sysB.countBonds(), 468)
		
	SDFile infileC(BALL_TEST_DATA_PATH(buildBondsProcessor_test.sdf));
	System sysC;
	infileC >> sysC;
	Size results[] = {9, 9, 9, 9, 9, 8, 9, 20, 6, 18, 12, 24, 21, 22};
	Size i(0);
	for (MoleculeIterator mit = sysC.beginMolecule(); +mit; ++mit, i++)
	{
		mit->apply(bbp2);
		TEST_EQUAL(mit->countBonds(), results[i]);
	}

	bbp2.options[BuildBondsProcessor::Option::DELETE_EXISTING_BONDS] = "true";
	i=0;
	for (MoleculeIterator mit = sysC.beginMolecule(); +mit; ++mit, i++)
	{
		mit->apply(bbp2);
		TEST_EQUAL(mit->countBonds(), results[i]);
	}

	bbp2.options[BuildBondsProcessor::Option::REESTIMATE_BONDORDERS_RINGS] = "false";
	bbp2.options[BuildBondsProcessor::Option::DELETE_OVERESTIMATED_BONDS] = "false";

	i=0;
	Size double_results[] = {3, 3, 0, 0, 0, 0, 0, 0, 3, 0, 3, 0, 3, 4};
	for (MoleculeIterator mit = sysC.beginMolecule(); +mit; ++mit, i++)
	{
		mit->apply(bbp2);
		TEST_EQUAL(mit->countBonds(), results[i])

		Size num_double(0);
		AtomIterator ait;
		Atom::BondIterator bit;
		BALL_FOREACH_BOND(*mit, ait, bit)
		{
			if (bit->getOrder() == Bond::ORDER__DOUBLE)
			{
				++num_double;
			}
		}
		TEST_EQUAL(num_double, double_results[i])
	}

RESULT

CHECK(setBondLengths(const String& filename))
	BuildBondsProcessor bbp2;
	bbp2.setBondLengths("bond_lengths/bond_lengths.db");
	PDBFile infileA(BALL_TEST_DATA_PATH(ACE_test_A.pdb));
	System sysA;
	infileA >> sysA;
	sysA.apply(bbp2);
	TEST_EQUAL(sysA.countBonds(), 1666)
	TEST_EXCEPTION(Exception::FileNotFound, bbp2.setBondLengths("file_does_not_exist"));
RESULT

/////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////

END_TEST