Source Code for Module test_suite.unit_tests._prompt.test_noe

  1  ############################################################################### 
  2  #                                                                             # 
  3  # Copyright (C) 2008, 2010 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.noe import Noe 
 28  from relax_errors import RelaxNoneIntError, RelaxNoneStrError, RelaxStrError 
 29   
 30  # Unit test imports. 
 31  from data_types import DATA_TYPES 
 32   
 33   
 34 -class Test_noe(TestCase): 
 35      """Unit tests for the functions of the 'prompt.noe' module.""" 
 36   
 37      # Instantiate the user function class. 
 38      noe_fns = Noe() 
 39   
 40   
 41 -    def test_read_restraints_argfail_file(self): 
 42          """The file arg test of the noe.read_restraints() user function.""" 
 43   
 44          # Loop over the data types. 
 45          for data in DATA_TYPES: 
 46              # Catch the str argument, and skip it. 
 47              if data[0] == 'str': 
 48                  continue 
 49   
 50              # The argument test. 
 51              self.assertRaises(RelaxStrError, self.noe_fns.read_restraints, file=data[1]) 
 52   
 53   
 54 -    def test_read_restraints_argfail_dir(self): 
 55          """The dir arg test of the noe.read_restraints() user function.""" 
 56   
 57          # Loop over the data types. 
 58          for data in DATA_TYPES: 
 59              # Catch the None and str arguments, and skip them. 
 60              if data[0] == 'None' or data[0] == 'str': 
 61                  continue 
 62   
 63              # The argument test. 
 64              self.assertRaises(RelaxNoneStrError, self.noe_fns.read_restraints, file='noes', dir=data[1]) 
 65   
 66   
 67 -    def test_read_restraints_argfail_proton1_col(self): 
 68          """The proton1_col arg test of the noe.read_restraints() user function.""" 
 69   
 70          # Loop over the data types. 
 71          for data in DATA_TYPES: 
 72              # Catch the None, int, and bin arguments, and skip them. 
 73              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
 74                  continue 
 75   
 76              # The argument test. 
 77              self.assertRaises(RelaxNoneIntError, self.noe_fns.read_restraints, file='noes', proton1_col=data[1]) 
 78   
 79   
 80 -    def test_read_restraints_argfail_proton2_col(self): 
 81          """The proton2_col arg test of the noe.read_restraints() user function.""" 
 82   
 83          # Loop over the data types. 
 84          for data in DATA_TYPES: 
 85              # Catch the None, int, and bin arguments, and skip them. 
 86              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
 87                  continue 
 88   
 89              # The argument test. 
 90              self.assertRaises(RelaxNoneIntError, self.noe_fns.read_restraints, file='noes', proton2_col=data[1]) 
 91   
 92   
 93 -    def test_read_restraints_argfail_lower_col(self): 
 94          """The lower_col arg test of the noe.read_restraints() user function.""" 
 95   
 96          # Loop over the data types. 
 97          for data in DATA_TYPES: 
 98              # Catch the int and bin arguments, and skip them. 
 99              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
100                  continue 
101   
102              # The argument test. 
103              self.assertRaises(RelaxNoneIntError, self.noe_fns.read_restraints, file='noes', lower_col=data[1]) 
104   
105   
106 -    def test_read_restraints_argfail_upper_col(self): 
107          """The upper_col arg test of the noe.read_restraints() user function.""" 
108   
109          # Loop over the data types. 
110          for data in DATA_TYPES: 
111              # Catch the int and bin arguments, and skip them. 
112              if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin': 
113                  continue 
114   
115              # The argument test. 
116              self.assertRaises(RelaxNoneIntError, self.noe_fns.read_restraints, file='noes', upper_col=data[1]) 
117   
118   
119 -    def test_read_restraints_argfail_sep(self): 
120          """The sep arg test of the noe.read_restraints() user function.""" 
121   
122          # Loop over the data types. 
123          for data in DATA_TYPES: 
124              # Catch the None and str arguments, and skip them. 
125              if data[0] == 'None' or data[0] == 'str': 
126                  continue 
127   
128              # The argument test. 
129              self.assertRaises(RelaxNoneStrError, self.noe_fns.read_restraints, file='noes', sep=data[1]) 
130   
131   
132 -    def test_spectrum_type_argfail_spectrum_type(self): 
133          """The spectrum_type arg test of the noe.spectrum_type() user function.""" 
134   
135          # Loop over the data types. 
136          for data in DATA_TYPES: 
137              # Catch the str argument, and skip it. 
138              if data[0] == 'str': 
139                  continue 
140   
141              # The argument test. 
142              self.assertRaises(RelaxStrError, self.noe_fns.spectrum_type, spectrum_type=data[1]) 
143   
144   
145 -    def test_spectrum_type_argfail_spectrum_id(self): 
146          """The spectrum_id arg test of the noe.spectrum_type() user function.""" 
147   
148          # Loop over the data types. 
149          for data in DATA_TYPES: 
150              # Catch the str argument, and skip it. 
151              if data[0] == 'str': 
152                  continue 
153   
154              # The argument test. 
155              self.assertRaises(RelaxStrError, self.noe_fns.spectrum_type, spectrum_type='x', spectrum_id=data[1]) 
156