Source Code for Module test_suite.unit_tests._generic_fns.test_diffusion_tensor

 1  ############################################################################### 
 2  #                                                                             # 
 3  # Copyright (C) 2007 Edward d'Auvergne                                        # 
 4  #                                                                             # 
 5  # This file is part of the program relax.                                     # 
 6  #                                                                             # 
 7  # relax is free software; you can redistribute it and/or modify               # 
 8  # it under the terms of the GNU General Public License as published by        # 
 9  # the Free Software Foundation; either version 2 of the License, or           # 
10  # (at your option) any later version.                                         # 
11  #                                                                             # 
12  # relax is distributed in the hope that it will be useful,                    # 
13  # but WITHOUT ANY WARRANTY; without even the implied warranty of              # 
14  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               # 
15  # GNU General Public License for more details.                                # 
16  #                                                                             # 
17  # You should have received a copy of the GNU General Public License           # 
18  # along with relax; if not, write to the Free Software                        # 
19  # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA   # 
20  #                                                                             # 
21  ############################################################################### 
22   
23  # Python module imports. 
24  from unittest import TestCase 
25   
26  # relax module imports. 
27  from generic_fns import diffusion_tensor 
28  from relax_errors import RelaxStrError 
29  from test_suite.unit_tests.diffusion_tensor_testing_base import Diffusion_tensor_base_class 
30   
31   
32   
33 -class Test_diffusion_tensor(Diffusion_tensor_base_class, TestCase): 
34      """Unit tests for the functions of the 'generic_fns.diffusion_tensor' module.""" 
35   
36      # Place the generic_fns.diffusion_tensor module into the class namespace. 
37      diffusion_tensor_fns = diffusion_tensor 
38   
39   
40 -    def test_return_data_name(self): 
41          """The returning of diffusion tensor parameter names. 
42   
43          The function tested is generic_fns.diffusion_tensor.return_data_name(). 
44          """ 
45   
46          # Test the return of diffusion tensor Eigenvalue components. 
47          self.assertEqual(self.diffusion_tensor_fns.return_data_name('tm'), 'tm') 
48          self.assertEqual(self.diffusion_tensor_fns.return_data_name('diso'), 'Diso') 
49          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Diso'), 'Diso') 
50          self.assertEqual(self.diffusion_tensor_fns.return_data_name('da'), 'Da') 
51          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Da'), 'Da') 
52          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dr'), 'Dr') 
53          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dr'), 'Dr') 
54          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dx'), 'Dx') 
55          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dx'), 'Dx') 
56          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dy'), 'Dy') 
57          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dy'), 'Dy') 
58          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dz'), 'Dz') 
59          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dz'), 'Dz') 
60          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dpar'), 'Dpar') 
61          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dpar'), 'Dpar') 
62          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dper'), 'Dper') 
63          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dper'), 'Dper') 
64          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dratio'), 'Dratio') 
65          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dratio'), 'Dratio') 
66   
67          # Test the return of Euler and spherical angles. 
68          self.assertEqual(self.diffusion_tensor_fns.return_data_name('a'), 'alpha') 
69          self.assertEqual(self.diffusion_tensor_fns.return_data_name('alpha'), 'alpha') 
70          self.assertEqual(self.diffusion_tensor_fns.return_data_name('b'), 'beta') 
71          self.assertEqual(self.diffusion_tensor_fns.return_data_name('beta'), 'beta') 
72          self.assertEqual(self.diffusion_tensor_fns.return_data_name('g'), 'gamma') 
73          self.assertEqual(self.diffusion_tensor_fns.return_data_name('gamma'), 'gamma') 
74          self.assertEqual(self.diffusion_tensor_fns.return_data_name('theta'), 'theta') 
75          self.assertEqual(self.diffusion_tensor_fns.return_data_name('phi'), 'phi') 
76   
77          # Test a few things which should fail. 
78          self.assertEqual(self.diffusion_tensor_fns.return_data_name('c'), None) 
79          self.assertEqual(self.diffusion_tensor_fns.return_data_name('7'), None) 
80          self.assertRaises(RelaxStrError, self.diffusion_tensor_fns.return_data_name, 7) 
81          self.assertEqual(self.diffusion_tensor_fns.return_data_name('tmm'), None) 
82