Source Code for Package test_suite.system_tests

  1  ############################################################################### 
  2  #                                                                             # 
  3  # Copyright (C) 2006-2012 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  # Package docstring. 
 23  """The relax system/functional tests.""" 
 24   
 25  # Python module imports. 
 26  from re import search 
 27  from unittest import TestSuite 
 28   
 29  # relax module imports. 
 30  from relax_errors import RelaxError 
 31   
 32  # relax system/functional test module imports. 
 33  from test_suite.relax_test_loader import RelaxTestLoader as TestLoader 
 34  from test_suite.system_tests.align_tensor import Align_tensor 
 35  from test_suite.system_tests.angles import Angles 
 36  from test_suite.system_tests.bmrb import Bmrb 
 37  from test_suite.system_tests.bruker import Bruker 
 38  from test_suite.system_tests.consistency_tests import Ct 
 39  from test_suite.system_tests.dasha import Dasha 
 40  from test_suite.system_tests.diffusion_tensor import Diffusion_tensor 
 41  from test_suite.system_tests.frame_order import Frame_order 
 42  from test_suite.system_tests.generic import Generic 
 43  from test_suite.system_tests.grace import Grace 
 44  from test_suite.system_tests.interatomic import Interatomic 
 45  from test_suite.system_tests.jw_mapping import Jw 
 46  from test_suite.system_tests.load_spins import Load_spins 
 47  from test_suite.system_tests.model_elimination import Modelim 
 48  from test_suite.system_tests.model_free import Mf 
 49  from test_suite.system_tests.model_selection import Modsel 
 50  from test_suite.system_tests.mol_res_spin import Mol_res_spin 
 51  from test_suite.system_tests.n_state_model import N_state_model 
 52  from test_suite.system_tests.noe import Noe 
 53  from test_suite.system_tests.noe_restraints import Noe_restraints 
 54  from test_suite.system_tests.palmer import Palmer 
 55  from test_suite.system_tests.pcs import Pcs 
 56  from test_suite.system_tests.peak_lists import Peak_lists 
 57  from test_suite.system_tests.pipes import Pipes 
 58  from test_suite.system_tests.rdc import Rdc 
 59  from test_suite.system_tests.relax_data import Relax_data 
 60  from test_suite.system_tests.relax_fit import Relax_fit 
 61  from test_suite.system_tests.results import Results 
 62  from test_suite.system_tests.selection import Selection 
 63  from test_suite.system_tests.sequence import Sequence 
 64  from test_suite.system_tests.state import State 
 65  from test_suite.system_tests.structure import Structure 
 66  from test_suite.system_tests.unit_vectors import Unit_vectors 
 67   
 68   
 69  __all__ = ['align_tensor', 
 70             'angles', 
 71             'bmrb', 
 72             'bruker', 
 73             'consistency_tests', 
 74             'dasha' 
 75             'diffusion_tensor', 
 76             'frame_order', 
 77             'generic', 
 78             'grace', 
 79             'interatomic', 
 80             'jw_mapping', 
 81             'load_spins', 
 82             'model_elimination', 
 83             'model_free', 
 84             'model_selection', 
 85             'n_state_model', 
 86             'noe', 
 87             'noe_restraints', 
 88             'palmer', 
 89             'pcs' 
 90             'peak_lists' 
 91             'pipes', 
 92             'rdc', 
 93             'relax_data', 
 94             'relax_fit', 
 95             'results', 
 96             'selection', 
 97             'sequence', 
 98             'state', 
 99             'structure', 
100             'unit_vectors'] 
101   
102   
103 -class System_test_runner: 
104      """Class for executing all of the system/functional tests.""" 
105   
106 -    def run(self, tests=None, runner=None): 
107          """Run the system/functional tests. 
108   
109          The system test list should be something like ['N_state_model.test_stereochem_analysis'].  The first part is the imported test case class, the second is the specific test. 
110   
111   
112          @keyword tests:     The list of system tests to preform. 
113          @type tests:        list of str 
114          @keyword runner:    A test runner such as TextTestRunner.  For an example of how to write a test runner see the python documentation for TextTestRunner in the python source. 
115          @type runner:       Test runner instance (TextTestRunner, BaseGUITestRunner subclass, etc.) 
116          """ 
117   
118          # Create an array of test suites (add your new TestCase classes here). 
119          suite_array = [] 
120   
121          # Specific tests. 
122          for test in tests: 
123              # The entire test class. 
124              if not search('\.', test): 
125                  # Check that the class exists. 
126                  if test not in globals(): 
127                      raise RelaxError("The system test class '%s' does not exist." % test) 
128   
129                  # The uninstantiated class object. 
130                  obj = globals()[test] 
131   
132                  # Add the tests. 
133                  suite_array.append(TestLoader().loadTestsFromTestCase(obj)) 
134   
135              # Single system test. 
136              else: 
137                  # Split. 
138                  row = test.split('.') 
139   
140                  # Check. 
141                  if len(row) != 2: 
142                      raise RelaxError("The test '%s' is not in the correct format.  It should consist of the test case class, a dot, and the specific test." % test) 
143   
144                  # Unpack. 
145                  class_name, test_name = row 
146   
147                  # Get the class object. 
148                  obj = globals()[class_name] 
149   
150                  # Add the test. 
151                  suite_array.append(TestLoader().loadTestsFromNames([test_name], obj)) 
152   
153          # All tests. 
154          if not tests: 
155              suite_array.append(TestLoader().loadTestsFromTestCase(Align_tensor)) 
156              suite_array.append(TestLoader().loadTestsFromTestCase(Bmrb)) 
157              suite_array.append(TestLoader().loadTestsFromTestCase(Bruker)) 
158              suite_array.append(TestLoader().loadTestsFromTestCase(Angles)) 
159              suite_array.append(TestLoader().loadTestsFromTestCase(Ct)) 
160              suite_array.append(TestLoader().loadTestsFromTestCase(Dasha)) 
161              suite_array.append(TestLoader().loadTestsFromTestCase(Diffusion_tensor)) 
162              suite_array.append(TestLoader().loadTestsFromTestCase(Frame_order)) 
163              suite_array.append(TestLoader().loadTestsFromTestCase(Generic)) 
164              suite_array.append(TestLoader().loadTestsFromTestCase(Grace)) 
165              suite_array.append(TestLoader().loadTestsFromTestCase(Interatomic)) 
166              suite_array.append(TestLoader().loadTestsFromTestCase(Jw)) 
167              suite_array.append(TestLoader().loadTestsFromTestCase(Load_spins)) 
168              suite_array.append(TestLoader().loadTestsFromTestCase(Modelim)) 
169              suite_array.append(TestLoader().loadTestsFromTestCase(Mf)) 
170              suite_array.append(TestLoader().loadTestsFromTestCase(Modsel)) 
171              suite_array.append(TestLoader().loadTestsFromTestCase(Mol_res_spin)) 
172              suite_array.append(TestLoader().loadTestsFromTestCase(N_state_model)) 
173              suite_array.append(TestLoader().loadTestsFromTestCase(Noe)) 
174              suite_array.append(TestLoader().loadTestsFromTestCase(Noe_restraints)) 
175              suite_array.append(TestLoader().loadTestsFromTestCase(Palmer)) 
176              suite_array.append(TestLoader().loadTestsFromTestCase(Pcs)) 
177              suite_array.append(TestLoader().loadTestsFromTestCase(Peak_lists)) 
178              suite_array.append(TestLoader().loadTestsFromTestCase(Pipes)) 
179              suite_array.append(TestLoader().loadTestsFromTestCase(Rdc)) 
180              suite_array.append(TestLoader().loadTestsFromTestCase(Relax_data)) 
181              suite_array.append(TestLoader().loadTestsFromTestCase(Relax_fit)) 
182              suite_array.append(TestLoader().loadTestsFromTestCase(Results)) 
183              suite_array.append(TestLoader().loadTestsFromTestCase(Selection)) 
184              suite_array.append(TestLoader().loadTestsFromTestCase(Sequence)) 
185              suite_array.append(TestLoader().loadTestsFromTestCase(State)) 
186              suite_array.append(TestLoader().loadTestsFromTestCase(Structure)) 
187              suite_array.append(TestLoader().loadTestsFromTestCase(Unit_vectors)) 
188   
189          # Group all tests together. 
190          full_suite = TestSuite(suite_array) 
191   
192          # Run the test suite. 
193          results = runner.run(full_suite) 
194   
195          # Return the status of the tests. 
196          return results.wasSuccessful() 
197