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
|
#!/usr/bin/env python
"""
This file tests vtk.vtkMolecule, and verifies that atoms/bonds are added.
"""
import sys
import vtk
from vtk.test import Testing
class TestMolecule(Testing.vtkTest):
def testCreation(self):
"Testing if molecules can be created/modified."
mol = vtk.vtkMolecule()
self.assertEqual(mol.GetNumberOfAtoms(), 0, "Number of atoms incorrect")
self.assertEqual(mol.GetNumberOfBonds(), 0, "Number of atoms incorrect")
h1 = mol.AppendAtom(1, 0.0, 0.0, -0.5)
h2 = mol.AppendAtom(1, 0.0, 0.0, 0.5)
b = mol.AppendBond(h1, h2, 1)
self.assertEqual(mol.GetNumberOfAtoms(), 2, "Number of atoms incorrect")
self.assertEqual(mol.GetNumberOfBonds(), 1, "Number of atoms incorrect")
if __name__ == "__main__":
Testing.main([(TestMolecule, 'test')])
|
