1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23 from lib.structure import pdb_read
24 from test_suite.unit_tests.base_classes import UnitTestCase
25
26
28 """Unit tests for the functions of the 'lib.structure.pdb_read' module."""
29
31 """Test the lib.structure.pdb_read.atom() function."""
32
33
34 record = pdb_read.atom('ATOM 158 CG GLU 11 9.590 -1.041 -11.596 1.00 0.00 C ')
35
36
37 self.assertEqual(record[0], 'ATOM')
38 self.assertEqual(record[1], 158)
39 self.assertEqual(record[2], 'CG')
40 self.assertEqual(record[3], None)
41 self.assertEqual(record[4], 'GLU')
42 self.assertEqual(record[5], None)
43 self.assertEqual(record[6], 11)
44 self.assertEqual(record[7], None)
45 self.assertEqual(record[8], 9.59)
46 self.assertEqual(record[9], -1.041)
47 self.assertEqual(record[10], -11.596)
48 self.assertEqual(record[11], 1.0)
49 self.assertEqual(record[12], 0.0)
50 self.assertEqual(record[13], 'C')
51 self.assertEqual(record[14], None)
52
53
55 """Test the lib.structure.pdb_read.helix() function."""
56
57
58 record = pdb_read.helix('HELIX 1 H1 ILE A 23 GLU A 34 1 12 ')
59
60
61 self.assertEqual(record[0], 'HELIX')
62 self.assertEqual(record[1], 1)
63 self.assertEqual(record[2], 'H1')
64 self.assertEqual(record[3], 'ILE')
65 self.assertEqual(record[4], 'A')
66 self.assertEqual(record[5], 23)
67 self.assertEqual(record[6], None)
68 self.assertEqual(record[7], 'GLU')
69 self.assertEqual(record[8], 'A')
70 self.assertEqual(record[9], 34)
71 self.assertEqual(record[10], None)
72 self.assertEqual(record[11], 1)
73 self.assertEqual(record[12], None)
74 self.assertEqual(record[13], 12)
75
76
78 """Test the lib.structure.pdb_read.sheet() function."""
79
80
81 record = pdb_read.sheet('SHEET 1 BET 5 GLY A 10 VAL A 17 0 ')
82
83
84 self.assertEqual(record[0], 'SHEET')
85 self.assertEqual(record[1], 1)
86 self.assertEqual(record[2], 'BET')
87 self.assertEqual(record[3], 5)
88 self.assertEqual(record[4], 'GLY')
89 self.assertEqual(record[5], 'A')
90 self.assertEqual(record[6], 10)
91 self.assertEqual(record[7], None)
92 self.assertEqual(record[8], 'VAL')
93 self.assertEqual(record[9], 'A')
94 self.assertEqual(record[10], 17)
95 self.assertEqual(record[11], None)
96 self.assertEqual(record[12], 0)
97 self.assertEqual(record[13], None)
98 self.assertEqual(record[14], None)
99 self.assertEqual(record[15], None)
100 self.assertEqual(record[16], None)
101 self.assertEqual(record[17], None)
102 self.assertEqual(record[18], None)
103 self.assertEqual(record[19], None)
104 self.assertEqual(record[20], None)
105 self.assertEqual(record[21], None)
106 self.assertEqual(record[22], None)
107