1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23 from data_store import Relax_data_store; ds = Relax_data_store()
24 from pipe_control import selection
25 from test_suite.unit_tests.base_classes import UnitTestCase
26
27
29 """Unit tests for the functions of the 'pipe_control.selection' module."""
30
32 """Set up some residues and spins for testing their selection and deselection."""
33
34
35 ds.add(pipe_name='orig', pipe_type='mf')
36
37
38 cdp.mol[0].name = 'Ap4Aase'
39
40
41 cdp.mol.add_item(mol_name='RNA')
42
43
44 cdp.mol[0].res[0].num = 1
45 cdp.mol[0].res.add_item(res_num=2, res_name='Glu')
46 cdp.mol[0].res.add_item(res_num=4, res_name='Pro')
47
48
49 cdp.mol[0].res[0].spin[0].name = 'NH'
50 cdp.mol[0].res[0].spin[0].num = 60
51 cdp.mol[0].res[1].spin[0].name = 'NH'
52 cdp.mol[0].res[1].spin[0].num = 63
53
54
55 cdp.mol[1].res[0].num = -5
56 cdp.mol[1].res.add_item(res_num=-4)
57
58
59 cdp.mol[1].res[0].spin[0].name = 'C8'
60 cdp.mol[1].res[1].spin[0].name = 'C8'
61 cdp.mol[1].res[0].spin.add_item(spin_name='N5')
62 cdp.mol[1].res[1].spin.add_item(spin_name='N5')
63
64
65 cdp.mol[0].res[0].spin[0].select = 0
66 cdp.mol[0].res[2].spin[0].select = 0
67 cdp.mol[1].res[0].spin[0].select = 0
68 cdp.mol[1].res[1].spin[1].select = 0
69
70
72 """Test spin system selection reversal.
73
74 The function tested is pipe_control.selection.reverse().
75 """
76
77
78 selection.reverse()
79
80
81 self.assertEqual(cdp.mol[0].res[0].spin[0].select, 1)
82 self.assertEqual(cdp.mol[0].res[1].spin[0].select, 0)
83 self.assertEqual(cdp.mol[0].res[2].spin[0].select, 1)
84 self.assertEqual(cdp.mol[1].res[0].spin[0].select, 1)
85 self.assertEqual(cdp.mol[1].res[0].spin[1].select, 0)
86 self.assertEqual(cdp.mol[1].res[1].spin[0].select, 0)
87 self.assertEqual(cdp.mol[1].res[1].spin[1].select, 1)
88