Source Code for Module test_suite.unit_tests._prompt.test_spin

  1  ############################################################################### 
  2  #                                                                             # 
  3  # Copyright (C) 2007-2012 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 prompt.interpreter import Interpreter 
 28  from relax_errors import RelaxIntError, RelaxListStrError, RelaxNoneIntError, RelaxNoneStrError, RelaxStrError 
 29  from test_suite.unit_tests.spin_testing_base import Spin_base_class 
 30   
 31  # Unit test imports. 
 32  from data_types import DATA_TYPES 
 33   
 34   
 35 -class Test_spin(Spin_base_class, TestCase): 
 36      """Unit tests for the functions of the 'prompt.spin' module.""" 
 37   
 38 -    def __init__(self, methodName=None): 
 39          """Set up the test case class for the system tests.""" 
 40   
 41          # Execute the base __init__ methods. 
 42          super(Test_spin, self).__init__(methodName) 
 43   
 44          # Load the interpreter. 
 45          self.interpreter = Interpreter(show_script=False, quit=False, raise_relax_error=True) 
 46          self.interpreter.populate_self() 
 47          self.interpreter.on(verbose=False) 
 48   
 49          # Alias the user function class. 
 50          self.spin_fns = self.interpreter.spin 
 51   
 52   
 53 -    def test_copy_argfail_pipe_from(self): 
 54          """Test the proper failure of the spin.copy() user function for the pipe_from argument.""" 
 55   
 56          # Loop over the data types. 
 57          for data in DATA_TYPES: 
 58              # Catch the None and str arguments, and skip them. 
 59              if data[0] == 'None' or data[0] == 'str': 
 60                  continue 
 61   
 62              # The argument test. 
 63              self.assertRaises(RelaxNoneStrError, self.spin_fns.copy, pipe_from=data[1], spin_from='#Old mol:1@111', spin_to='#Old mol:2') 
 64   
 65   
 66 -    def test_copy_argfail_spin_from(self): 
 67          """Test the proper failure of the spin.copy() user function for the spin_from argument.""" 
 68   
 69          # Loop over the data types. 
 70          for data in DATA_TYPES: 
 71              # Catch the str argument, and skip it. 
 72              if data[0] == 'str': 
 73                  continue 
 74   
 75              # The argument test. 
 76              self.assertRaises(RelaxStrError, self.spin_fns.copy, spin_from=data[1], spin_to='#Old mol:2') 
 77   
 78   
 79 -    def test_copy_argfail_pipe_to(self): 
 80          """Test the proper failure of the spin.copy() user function for the pipe_to argument.""" 
 81   
 82          # Loop over the data types. 
 83          for data in DATA_TYPES: 
 84              # Catch the None and str arguments, and skip them. 
 85              if data[0] == 'None' or data[0] == 'str': 
 86                  continue 
 87   
 88              # The argument test. 
 89              self.assertRaises(RelaxNoneStrError, self.spin_fns.copy, pipe_to=data[1], spin_from='#Old mol:1@111', spin_to='#Old mol:2') 
 90   
 91   
 92 -    def test_copy_argfail_spin_to(self): 
 93          """Test the proper failure of the spin.copy() user function for the spin_to argument.""" 
 94   
 95          # Loop over the data types. 
 96          for data in DATA_TYPES: 
 97              # Catch the None and str arguments, and skip them. 
 98              if data[0] == 'None' or  data[0] == 'str': 
 99                  continue 
100   
101              # The argument test. 
102              self.assertRaises(RelaxNoneStrError, self.spin_fns.copy, spin_from='#Old mol:1@111', spin_to=data[1]) 
103   
104   
105 -    def test_create_argfail_spin_num(self): 
106          """Test the proper failure of the spin.create() user function for the spin_num argument.""" 
107   
108          # Loop over the data types. 
109          for data in DATA_TYPES: 
110              # Catch the int and bin arguments, and skip them. 
111              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
112                  continue 
113   
114              # The argument test. 
115              self.assertRaises(RelaxNoneIntError, self.spin_fns.create, spin_num=data[1], spin_name='NH') 
116   
117   
118 -    def test_create_argfail_spin_name(self): 
119          """Test the proper failure of the spin.create() user function for the spin_name argument.""" 
120   
121          # Loop over the data types. 
122          for data in DATA_TYPES: 
123              # Catch the None and str arguments, and skip them. 
124              if data[0] == 'None' or data[0] == 'str': 
125                  continue 
126   
127              # The argument test. 
128              self.assertRaises(RelaxNoneStrError, self.spin_fns.create, spin_name=data[1], spin_num=1) 
129   
130   
131 -    def test_create_argfail_res_num(self): 
132          """Test the proper failure of the spin.create() user function for the res_num argument.""" 
133   
134          # Loop over the data types. 
135          for data in DATA_TYPES: 
136              # Catch the int and bin arguments, and skip them. 
137              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
138                  continue 
139   
140              # The argument test. 
141              self.assertRaises(RelaxNoneIntError, self.spin_fns.create, res_num=data[1], spin_name='NH') 
142   
143   
144 -    def test_create_argfail_res_name(self): 
145          """Test the proper failure of the spin.create() user function for the res_name argument.""" 
146   
147          # Loop over the data types. 
148          for data in DATA_TYPES: 
149              # Catch the None and str arguments, and skip them. 
150              if data[0] == 'None' or data[0] == 'str': 
151                  continue 
152   
153              # The argument test. 
154              self.assertRaises(RelaxNoneStrError, self.spin_fns.create, res_name=data[1], spin_num=1, spin_name='NH') 
155   
156   
157 -    def test_create_argfail_mol_name(self): 
158          """Test the proper failure of the spin.create() user function for the mol_name argument.""" 
159   
160          # Loop over the data types. 
161          for data in DATA_TYPES: 
162              # Catch the None and str arguments, and skip them. 
163              if data[0] == 'None' or data[0] == 'str': 
164                  continue 
165   
166              # The argument test. 
167              self.assertRaises(RelaxNoneStrError, self.spin_fns.create, mol_name=data[1], spin_num=1, spin_name='NH') 
168   
169   
170 -    def test_create_pseudo_argfail_spin_name(self): 
171          """The spin_name arg test of the spin.create_pseudo() user function.""" 
172   
173          # Loop over the data types. 
174          for data in DATA_TYPES: 
175              # Catch the None and str arguments, and skip them. 
176              if data[0] == 'str': 
177                  continue 
178   
179              # The argument test. 
180              self.assertRaises(RelaxStrError, self.spin_fns.create_pseudo, spin_name=data[1]) 
181   
182   
183 -    def test_create_pseudo_argfail_spin_num(self): 
184          """The spin_num arg test of the spin.create_pseudo() user function.""" 
185   
186          # Loop over the data types. 
187          for data in DATA_TYPES: 
188              # Catch the int and bin arguments, and skip them. 
189              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
190                  continue 
191   
192              # The argument test. 
193              self.assertRaises(RelaxNoneIntError, self.spin_fns.create_pseudo, spin_num=data[1], spin_name='Q') 
194   
195   
196 -    def test_create_pseudo_argfail_res_id(self): 
197          """The res_id arg test of the spin.create_pseudo() user function.""" 
198   
199          # Loop over the data types. 
200          for data in DATA_TYPES: 
201              # Catch the None and str arguments, and skip them. 
202              if data[0] == 'None' or data[0] == 'str': 
203                  continue 
204   
205              # The argument test. 
206              self.assertRaises(RelaxNoneStrError, self.spin_fns.create_pseudo, res_id=data[1], spin_name='Q') 
207   
208   
209 -    def test_create_pseudo_argfail_members(self): 
210          """The members arg test of the spin.create_pseudo() user function.""" 
211   
212          # Loop over the data types. 
213          for data in DATA_TYPES: 
214              # Catch the str list argument, and skip it. 
215              if data[0] == 'str list': 
216                  continue 
217   
218              # The argument test. 
219              self.assertRaises(RelaxListStrError, self.spin_fns.create_pseudo, members=data[1], spin_name='Q') 
220   
221   
222 -    def test_create_pseudo_argfail_averaging(self): 
223          """The averaging arg test of the spin.create_pseudo() user function.""" 
224   
225          # Loop over the data types. 
226          for data in DATA_TYPES: 
227              # Catch the str arguments, and skip them. 
228              if data[0] == 'str': 
229                  continue 
230   
231              # The argument test. 
232              self.assertRaises(RelaxStrError, self.spin_fns.create_pseudo, averaging=data[1], spin_name='Q', members=['x']) 
233   
234   
235 -    def test_delete_argfail_spin_id(self): 
236          """Test the proper failure of the spin.delete() user function for the spin_id argument.""" 
237   
238          # Loop over the data types. 
239          for data in DATA_TYPES: 
240              # Catch the str arguments, and skip them. 
241              if data[0] == 'str': 
242                  continue 
243   
244              # The argument test. 
245              self.assertRaises(RelaxStrError, self.spin_fns.delete, spin_id=data[1]) 
246   
247   
248 -    def test_display_argfail_spin_id(self): 
249          """Test the proper failure of the spin.display() user function for the spin_id argument.""" 
250   
251          # Loop over the data types. 
252          for data in DATA_TYPES: 
253              # Catch the None and str arguments, and skip them. 
254              if data[0] == 'None' or data[0] == 'str': 
255                  continue 
256   
257              # The argument test. 
258              self.assertRaises(RelaxNoneStrError, self.spin_fns.display, spin_id=data[1]) 
259   
260   
261 -    def test_name_argfail_spin_id(self): 
262          """Test the proper failure of the spin.name() user function for the spin_id argument.""" 
263   
264          # Loop over the data types. 
265          for data in DATA_TYPES: 
266              # Catch the None and str arguments, and skip them. 
267              if data[0] == 'None' or data[0] == 'str': 
268                  continue 
269   
270              # The argument test. 
271              self.assertRaises(RelaxNoneStrError, self.spin_fns.name, name='N', spin_id=data[1]) 
272   
273   
274 -    def test_name_argfail_name(self): 
275          """Test the proper failure of the spin.name() user function for the name argument.""" 
276   
277          # Loop over the data types. 
278          for data in DATA_TYPES: 
279              # Catch the str arguments, and skip them. 
280              if data[0] == 'str': 
281                  continue 
282   
283              # The argument test. 
284              self.assertRaises(RelaxStrError, self.spin_fns.name, name=data[1]) 
285   
286   
287 -    def test_number_argfail_spin_id(self): 
288          """Test the proper failure of the spin.number() user function for the spin_id argument.""" 
289   
290          # Loop over the data types. 
291          for data in DATA_TYPES: 
292              # Catch the None and str arguments, and skip them. 
293              if data[0] == 'None' or data[0] == 'str': 
294                  continue 
295   
296              # The argument test. 
297              self.assertRaises(RelaxNoneStrError, self.spin_fns.number, spin_id=data[1]) 
298   
299   
300 -    def test_number_argfail_number(self): 
301          """Test the proper failure of the spin.number() user function for the number argument.""" 
302   
303          # Loop over the data types. 
304          for data in DATA_TYPES: 
305              # Catch the None, int and bin arguments, and skip them. 
306              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
307                  continue 
308   
309              # The argument test. 
310              self.assertRaises(RelaxNoneIntError, self.spin_fns.number, spin_id='@111', number=data[1]) 
311