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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
'''
Testing: pymol.editor
'''
from pymol import cmd, testing, stored, editor
class TestEditorFab(testing.PyMOLTestCase):
'''
fab(input, name=None, mode='peptide', resi=1, chain='', segi='',
state=-1, dir=1, hydro=-1, ss=0, async=-1, quiet=1)
The only valid mode is 'peptide'.
'''
def testSimple(self):
cmd.fab('AA')
v = cmd.get_object_list()
self.assertEqual(v, ['obj01'])
def testAdvanced(self):
cmd.fab('A/123/ ADC B/234/ AFCD')
v = cmd.get_chains()
self.assertEqual(v, ['A', 'B'])
cmd.iterate('last chain B', 'stored.v = (resv, resn)')
self.assertEqual(stored.v, (237, 'ASP'))
def testIdentifiers(self):
seq = 'ACD'
segi = 'foo'
chain = 'F'
resv = 10
cmd.fab(seq, 'm1', 'peptide', resv, chain, segi)
cmd.iterate('first m1', 'stored.v = (segi, chain, resv, resn)')
self.assertEqual(stored.v, (segi, chain, resv, 'ALA'))
cmd.iterate('last m1', 'stored.v = (segi, chain, resv, resn)')
self.assertEqual(stored.v, (segi, chain, resv + 2, 'ASP'))
v = cmd.get_fastastr().splitlines()[1]
self.assertEqual(v, seq)
def testDir(self):
seq = 'ACD'
resv = 5
cmd.fab(seq, 'm1', resi=resv, dir=-1)
# incentive needs sort, opensource is already sorted
cmd.sort()
cmd.iterate('first m1', 'stored.v = (resv, resn)')
self.assertEqual(stored.v, (resv - 2, 'ASP'))
@testing.foreach('AA', 'A')
def testHydro(self, seq):
cmd.fab(seq, hydro='0')
v = cmd.count_atoms('hydro')
self.assertEqual(0, v, 'failed to remove hydrogens on ' + seq)
@testing.foreach.zip(range(5))
def testSS(self, ss):
cmd.fab('AAAPAAA', 'm1', ss=ss)
|
