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.peak_lists import Peak_lists 
 56  from test_suite.system_tests.pipes import Pipes 
 57  from test_suite.system_tests.rdc import Rdc 
 58  from test_suite.system_tests.relax_data import Relax_data 
 59  from test_suite.system_tests.relax_fit import Relax_fit 
 60  from test_suite.system_tests.results import Results 
 61  from test_suite.system_tests.sequence import Sequence 
 62  from test_suite.system_tests.state import State 
 63  from test_suite.system_tests.structure import Structure 
 64  from test_suite.system_tests.unit_vectors import Unit_vectors 
 65   
 66   
 67  __all__ = ['align_tensor', 
 68             'angles', 
 69             'bmrb', 
 70             'bruker', 
 71             'consistency_tests', 
 72             'dasha' 
 73             'diffusion_tensor', 
 74             'frame_order', 
 75             'generic', 
 76             'grace', 
 77             'interatomic', 
 78             'jw_mapping', 
 79             'load_spins', 
 80             'model_elimination', 
 81             'model_free', 
 82             'model_selection', 
 83             'n_state_model', 
 84             'noe', 
 85             'noe_restraints', 
 86             'palmer', 
 87             'peak_lists' 
 88             'pipes', 
 89             'rdc', 
 90             'relax_data', 
 91             'relax_fit', 
 92             'results', 
 93             'sequence', 
 94             'state', 
 95             'structure', 
 96             'unit_vectors'] 
 97   
 98   
 99 -class System_test_runner: 
100      """Class for executing all of the system/functional tests.""" 
101   
102 -    def run(self, tests=None, runner=None): 
103          """Run the system/functional tests. 
104   
105          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. 
106   
107   
108          @keyword tests:     The list of system tests to preform. 
109          @type tests:        list of str 
110          @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. 
111          @type runner:       Test runner instance (TextTestRunner, BaseGUITestRunner subclass, etc.) 
112          """ 
113   
114          # Create an array of test suites (add your new TestCase classes here). 
115          suite_array = [] 
116   
117          # Specific tests. 
118          for test in tests: 
119              # The entire test class. 
120              if not search('\.', test): 
121                  # Check that the class exists. 
122                  if test not in globals(): 
123                      raise RelaxError("The system test class '%s' does not exist." % test) 
124   
125                  # The uninstantiated class object. 
126                  obj = globals()[test] 
127   
128                  # Add the tests. 
129                  suite_array.append(TestLoader().loadTestsFromTestCase(obj)) 
130   
131              # Single system test. 
132              else: 
133                  # Split. 
134                  row = test.split('.') 
135   
136                  # Check. 
137                  if len(row) != 2: 
138                      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) 
139   
140                  # Unpack. 
141                  class_name, test_name = row 
142   
143                  # Get the class object. 
144                  obj = globals()[class_name] 
145   
146                  # Add the test. 
147                  suite_array.append(TestLoader().loadTestsFromNames([test_name], obj)) 
148   
149          # All tests. 
150          if not tests: 
151              suite_array.append(TestLoader().loadTestsFromTestCase(Align_tensor)) 
152              suite_array.append(TestLoader().loadTestsFromTestCase(Bmrb)) 
153              suite_array.append(TestLoader().loadTestsFromTestCase(Bruker)) 
154              suite_array.append(TestLoader().loadTestsFromTestCase(Angles)) 
155              suite_array.append(TestLoader().loadTestsFromTestCase(Ct)) 
156              suite_array.append(TestLoader().loadTestsFromTestCase(Dasha)) 
157              suite_array.append(TestLoader().loadTestsFromTestCase(Diffusion_tensor)) 
158              suite_array.append(TestLoader().loadTestsFromTestCase(Frame_order)) 
159              suite_array.append(TestLoader().loadTestsFromTestCase(Generic)) 
160              suite_array.append(TestLoader().loadTestsFromTestCase(Grace)) 
161              suite_array.append(TestLoader().loadTestsFromTestCase(Interatomic)) 
162              suite_array.append(TestLoader().loadTestsFromTestCase(Jw)) 
163              suite_array.append(TestLoader().loadTestsFromTestCase(Load_spins)) 
164              suite_array.append(TestLoader().loadTestsFromTestCase(Modelim)) 
165              suite_array.append(TestLoader().loadTestsFromTestCase(Mf)) 
166              suite_array.append(TestLoader().loadTestsFromTestCase(Modsel)) 
167              suite_array.append(TestLoader().loadTestsFromTestCase(Mol_res_spin)) 
168              suite_array.append(TestLoader().loadTestsFromTestCase(N_state_model)) 
169              suite_array.append(TestLoader().loadTestsFromTestCase(Noe)) 
170              suite_array.append(TestLoader().loadTestsFromTestCase(Noe_restraints)) 
171              suite_array.append(TestLoader().loadTestsFromTestCase(Palmer)) 
172              suite_array.append(TestLoader().loadTestsFromTestCase(Peak_lists)) 
173              suite_array.append(TestLoader().loadTestsFromTestCase(Pipes)) 
174              suite_array.append(TestLoader().loadTestsFromTestCase(Rdc)) 
175              suite_array.append(TestLoader().loadTestsFromTestCase(Relax_data)) 
176              suite_array.append(TestLoader().loadTestsFromTestCase(Relax_fit)) 
177              suite_array.append(TestLoader().loadTestsFromTestCase(Results)) 
178              suite_array.append(TestLoader().loadTestsFromTestCase(Sequence)) 
179              suite_array.append(TestLoader().loadTestsFromTestCase(State)) 
180              suite_array.append(TestLoader().loadTestsFromTestCase(Structure)) 
181              suite_array.append(TestLoader().loadTestsFromTestCase(Unit_vectors)) 
182   
183          # Group all tests together. 
184          full_suite = TestSuite(suite_array) 
185   
186          # Run the test suite. 
187          results = runner.run(full_suite) 
188   
189          # Return the status of the tests. 
190          return results.wasSuccessful() 
191