2
"""Unit tests for the pdb parser.
4
from cogent.util.unit_test import TestCase, main
5
from cogent.core.entity import Structure
6
from cogent.parse.structure import FromFilenameStructureParser, FromFileStructureParser
8
__author__ = "Marcin Knight"
9
__copyright__ = "Copyright 2007-2009, The Cogent Project"
10
__credits__ = ["Marcin Cieslik"]
13
__maintainer__ = "Marcin Cieslik"
14
__email__ = "mpc4p@virginia.edu"
15
__status__ = "Production"
18
class structuresTests(TestCase):
19
"""Tests of cogent.parse.structure UI functions."""
21
def test_FromFilenameStructureParser(self):
22
structure = FromFilenameStructureParser('data/1LJO.pdb', 'pdb')
23
self.assertRaises(TypeError, FromFilenameStructureParser, open('data/1LJO.pdb'), 'pdb')
24
assert isinstance(structure, Structure)
26
def test_FromFileStructureParser(self):
27
structure = FromFileStructureParser(open('data/1LJO.pdb'), 'pdb')
28
assert isinstance(structure, Structure)
29
self.assertRaises(TypeError, FromFileStructureParser, 'data/1LJO.pdb', 'pdb')
30
assert isinstance(structure, Structure)
33
if __name__ == '__main__':