Source Code for Module test_suite.unit_tests._pipe_control.test_diffusion_tensor

 1  ############################################################################### 
 2  #                                                                             # 
 3  # Copyright (C) 2007-2013 Edward d'Auvergne                                   # 
 4  #                                                                             # 
 5  # This file is part of the program relax (http://www.nmr-relax.com).          # 
 6  #                                                                             # 
 7  # This program 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 3 of the License, or           # 
10  # (at your option) any later version.                                         # 
11  #                                                                             # 
12  # This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.       # 
19  #                                                                             # 
20  ############################################################################### 
21   
22  # Python module imports. 
23  from unittest import TestCase 
24   
25  # relax module imports. 
26  from pipe_control import diffusion_tensor 
27  from lib.errors import RelaxStrError 
28  from test_suite.unit_tests.diffusion_tensor_testing_base import Diffusion_tensor_base_class 
29   
30   
31   
32 -class Test_diffusion_tensor(Diffusion_tensor_base_class, TestCase): 
33      """Unit tests for the functions of the 'pipe_control.diffusion_tensor' module.""" 
34   
35      # Place the pipe_control.diffusion_tensor module into the class namespace. 
36      diffusion_tensor_fns = diffusion_tensor 
37   
38   
39 -    def test_return_data_name(self): 
40          """The returning of diffusion tensor parameter names. 
41   
42          The function tested is pipe_control.diffusion_tensor.return_data_name(). 
43          """ 
44   
45          # Test the return of diffusion tensor Eigenvalue components. 
46          self.assertEqual(self.diffusion_tensor_fns.return_data_name('tm'), 'tm') 
47          self.assertEqual(self.diffusion_tensor_fns.return_data_name('diso'), 'Diso') 
48          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Diso'), 'Diso') 
49          self.assertEqual(self.diffusion_tensor_fns.return_data_name('da'), 'Da') 
50          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Da'), 'Da') 
51          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dr'), 'Dr') 
52          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dr'), 'Dr') 
53          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dx'), 'Dx') 
54          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dx'), 'Dx') 
55          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dy'), 'Dy') 
56          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dy'), 'Dy') 
57          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dz'), 'Dz') 
58          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dz'), 'Dz') 
59          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dpar'), 'Dpar') 
60          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dpar'), 'Dpar') 
61          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dper'), 'Dper') 
62          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dper'), 'Dper') 
63          self.assertEqual(self.diffusion_tensor_fns.return_data_name('dratio'), 'Dratio') 
64          self.assertEqual(self.diffusion_tensor_fns.return_data_name('Dratio'), 'Dratio') 
65   
66          # Test the return of Euler and spherical angles. 
67          self.assertEqual(self.diffusion_tensor_fns.return_data_name('a'), 'alpha') 
68          self.assertEqual(self.diffusion_tensor_fns.return_data_name('alpha'), 'alpha') 
69          self.assertEqual(self.diffusion_tensor_fns.return_data_name('b'), 'beta') 
70          self.assertEqual(self.diffusion_tensor_fns.return_data_name('beta'), 'beta') 
71          self.assertEqual(self.diffusion_tensor_fns.return_data_name('g'), 'gamma') 
72          self.assertEqual(self.diffusion_tensor_fns.return_data_name('gamma'), 'gamma') 
73          self.assertEqual(self.diffusion_tensor_fns.return_data_name('theta'), 'theta') 
74          self.assertEqual(self.diffusion_tensor_fns.return_data_name('phi'), 'phi') 
75   
76          # Test a few things which should fail. 
77          self.assertEqual(self.diffusion_tensor_fns.return_data_name('c'), None) 
78          self.assertEqual(self.diffusion_tensor_fns.return_data_name('7'), None) 
79          self.assertRaises(RelaxStrError, self.diffusion_tensor_fns.return_data_name, 7) 
80          self.assertEqual(self.diffusion_tensor_fns.return_data_name('tmm'), None) 
81