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