File: AromaticityProcessor_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 (80 lines) | stat: -rw-r--r-- 1,679 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
// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
//
#include <BALL/CONCEPT/classTest.h>
#include <BALLTestConfig.h>

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

#include <BALL/QSAR/aromaticityProcessor.h>
#include <BALL/FORMAT/SDFile.h>
#include <BALL/KERNEL/system.h>
#include <BALL/KERNEL/atom.h>
#include <BALL/KERNEL/bond.h>
#include <BALL/KERNEL/forEach.h>
#include <BALL/KERNEL/molecule.h>
#include <BALL/KERNEL/PTE.h>

///////////////////////////
START_TEST(AromaticityProcessor)

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

using namespace BALL;

SDFile infile(BALL_TEST_DATA_PATH(descriptors_test.sdf));
System S;
infile >> S;
infile.close();

Molecule * molecule;
Size limit = S.countMolecules();

CHECK(AromaticityProcessor)	

	AromaticityProcessor ap;
	S.apply(ap);

	
	Size results[] = {0, 6, 0, 6, 6, 6, 0, 6, 0};
	for (Size i=0;i!=limit;++i)
	{
		molecule = S.getMolecule(i);

		// atoms
		Size num_aro = 0;
		AtomConstIterator a_it = molecule->beginAtom();
		for (;a_it!=molecule->endAtom();++a_it)
		{
			if(a_it->hasProperty("IsAromatic"))
			{
				if (a_it->getProperty("IsAromatic").getBool())
				{
					++num_aro;
				}
			}
		}
		TEST_EQUAL(num_aro, results[i])

		// bonds
		a_it = molecule->beginAtom();
		Atom::BondConstIterator b_it = a_it->beginBond();
		num_aro = 0;
		BALL_FOREACH_BOND(*molecule, a_it, b_it)
		{
			if (b_it->getOrder() == Bond::ORDER__AROMATIC)
			{
				++num_aro;
			}
		}
		TEST_EQUAL(num_aro, results[i])		
	}
		
RESULT


/////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////
END_TEST