Commit c98f0766 authored by Marcus D. Hanwell's avatar Marcus D. Hanwell
Browse files

Added a simple test to verify wrapped vtkMolecule

parent 9b868cf9
......@@ -15,4 +15,5 @@ vtk_add_test_python(
quadricCut.py,NO_RT
quadraticQuadDeriv.py,NO_VALID,NO_RT
TestICPTransform.py,NO_RT
TestMolecule.py,NO_VALID,NO_RT
)
#!/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')])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment