Package test_suite :: Package unit_tests :: Package _prompt :: Module test_pymol
[hide private]
[frames] | no frames]

Source Code for Module test_suite.unit_tests._prompt.test_pymol

  1  ############################################################################### 
  2  #                                                                             # 
  3  # Copyright (C) 2009-2014 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 prompt.interpreter import Interpreter 
 27  from lib.errors import RelaxNoneStrError, RelaxNoneListNumError, RelaxStrError 
 28   
 29  # Unit test imports. 
 30  from test_suite.unit_tests._prompt.data_types import DATA_TYPES 
 31   
 32   
33 -class Test_pymol(TestCase):
34 """Unit tests for the functions of the 'prompt.pymol' module.""" 35
36 - def __init__(self, methodName=None):
37 """Set up the test case class for the system tests.""" 38 39 # Execute the base __init__ methods. 40 super(Test_pymol, self).__init__(methodName) 41 42 # Load the interpreter. 43 self.interpreter = Interpreter(show_script=False, raise_relax_error=True) 44 self.interpreter.populate_self() 45 self.interpreter.on(verbose=False) 46 47 # Alias the user function class. 48 self.pymol_fns = self.interpreter.pymol
49 50
52 """The data_type arg test of the pymol.macro_apply() user function.""" 53 54 # Loop over the data types. 55 for data in DATA_TYPES: 56 # Catch the str arguments, and skip them. 57 if data[0] == 'str': 58 continue 59 60 # The argument test. 61 self.assertRaises(RelaxStrError, self.pymol_fns.macro_apply, data_type=data[1])
62 63
65 """The style arg test of the pymol.macro_apply() user function.""" 66 67 # Loop over the data types. 68 for data in DATA_TYPES: 69 # Catch the str arguments, and skip them. 70 if data[0] == 'str': 71 continue 72 73 # The argument test. 74 self.assertRaises(RelaxStrError, self.pymol_fns.macro_apply, data_type='a', style=data[1])
75 76
78 """The colour_start_name arg test of the pymol.macro_apply() user function.""" 79 80 # Loop over the data types. 81 for data in DATA_TYPES: 82 # Catch the None and str arguments, and skip them. 83 if data[0] == 'None' or data[0] == 'str': 84 continue 85 86 # The argument test. 87 self.assertRaises(RelaxNoneStrError, self.pymol_fns.macro_apply, data_type='a', style='x', colour_start_name=data[1])
88 89
91 """The colour_start_rgb arg test of the pymol.macro_apply() user function.""" 92 93 # Loop over the data types. 94 for data in DATA_TYPES: 95 # Catch the None and num list arguments, and skip them. 96 if data[0] == 'None' or ((data[0] == 'int list' or data[0] == 'float list' or data[0] == 'number list') and len(data[1]) == 3): 97 continue 98 99 # The argument test. 100 self.assertRaises(RelaxNoneListNumError, self.pymol_fns.macro_apply, data_type='a', style='x', colour_start_rgb=data[1])
101 102
104 """The colour_end_name arg test of the pymol.macro_apply() user function.""" 105 106 # Loop over the data types. 107 for data in DATA_TYPES: 108 # Catch the None and str arguments, and skip them. 109 if data[0] == 'None' or data[0] == 'str': 110 continue 111 112 # The argument test. 113 self.assertRaises(RelaxNoneStrError, self.pymol_fns.macro_apply, data_type='a', style='x', colour_end_name=data[1])
114 115
117 """The colour_end_rgb arg test of the pymol.macro_apply() user function.""" 118 119 # Loop over the data types. 120 for data in DATA_TYPES: 121 # Catch the None and num list arguments, and skip them. 122 if data[0] == 'None' or ((data[0] == 'int list' or data[0] == 'float list' or data[0] == 'number list') and len(data[1]) == 3): 123 continue 124 125 # The argument test. 126 self.assertRaises(RelaxNoneListNumError, self.pymol_fns.macro_apply, data_type='a', style='x', colour_end_rgb=data[1])
127 128
130 """The colour_list arg test of the pymol.macro_apply() user function.""" 131 132 # Loop over the data types. 133 for data in DATA_TYPES: 134 # Catch the None and str arguments, and skip them. 135 if data[0] == 'None' or data[0] == 'str': 136 continue 137 138 # The argument test. 139 self.assertRaises(RelaxNoneStrError, self.pymol_fns.macro_apply, data_type='a', style='x', colour_list=data[1])
140