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_write
24 from lib.io import DummyFileObject
25 from test_suite.unit_tests.base_classes import UnitTestCase
26
27
29 """Unit tests for the functions of the 'lib.structure.pdb_write' module."""
30
32 """Test the lib.structure.pdb_write.atom() function."""
33
34
35 file = DummyFileObject()
36
37
38 pdb_write.atom(file, serial=158, name='CG', res_name='GLU', res_seq='11', x=9.59, y=-1.041, z=-11.596, occupancy=1.0, temp_factor=0.0, element='C')
39
40
41 records = file.readlines()
42 actual = 'ATOM 158 CG GLU 11 9.590 -1.041 -11.596 1.00 0.00 C \n'
43 print(repr(records[0]))
44 print(repr(actual))
45 self.assertEqual(records[0], actual)
46
47
49 """Test the lib.structure.pdb_write.helix() function."""
50
51
52 file = DummyFileObject()
53
54
55 pdb_write.helix(file, ser_num=1, helix_id='H1', init_res_name='ILE', init_chain_id='A', init_seq_num=23, init_icode=None, end_res_name='GLU', end_chain_id='A', end_seq_num=34, end_icode=None, helix_class=1, comment=None, length=12)
56
57
58 records = file.readlines()
59 actual = 'HELIX 1 H1 ILE A 23 GLU A 34 1 12 \n'
60 print(repr(records[0]))
61 print(repr(actual))
62 self.assertEqual(records[0], actual)
63
64
66 """Test the lib.structure.pdb_write.het() function."""
67
68
69 file = DummyFileObject()
70
71
72 pdb_write.het(file, het_id='CA', chain_id='A', seq_num=1000, icode=None, num_het_atoms=1, text=None)
73
74
75 records = file.readlines()
76 actual = 'HET CA A1000 1 \n'
77 print(repr(records[0]))
78 print(repr(actual))
79 self.assertEqual(records[0], actual)
80
81
83 """Test the lib.structure.pdb_write.model() function."""
84
85
86 file = DummyFileObject()
87
88
89 pdb_write.model(file, serial=1)
90
91
92 records = file.readlines()
93 actual = 'MODEL 1 \n'
94 print(repr(records[0]))
95 print(repr(actual))
96 self.assertEqual(records[0], actual)
97
98
100 """Test the lib.structure.pdb_write.sheet() function."""
101
102
103 file = DummyFileObject()
104
105
106 pdb_write.sheet(file, strand=1, sheet_id='BET', num_strands=5, init_res_name='GLY', init_chain_id='A', init_seq_num=10, init_icode=None, end_res_name='VAL', end_chain_id='A', end_seq_num=17, end_icode=None, sense=0, cur_atom=None, cur_res_name=None, cur_chain_id=None, cur_res_seq=None, cur_icode=None, prev_atom=None, prev_res_name=None, prev_chain_id=None, prev_res_seq=None, prev_icode=None)
107
108
109 records = file.readlines()
110 actual = 'SHEET 1 BET 5 GLY A 10 VAL A 17 0 \n'
111 print(repr(records[0]))
112 print(repr(actual))
113 self.assertEqual(records[0], actual)
114