1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23 from unittest import TestCase
24
25
26 from prompt.interpreter import Interpreter
27 from lib.errors import RelaxIntError, RelaxNoneIntError, RelaxNoneIntListIntError, RelaxNoneStrError, RelaxNumError, RelaxStrError, RelaxStrListStrError
28
29
30 from test_suite.unit_tests._prompt.data_types import DATA_TYPES
31
32
34 """Unit tests for the functions of the 'prompt.spectrum' module."""
35
49
50
62
63
65 """The spectrum_id arg test of the spectrum.baseplane_rmsd() user function."""
66
67
68 for data in DATA_TYPES:
69
70 if data[0] == 'str':
71 continue
72
73
74 self.assertRaises(RelaxStrError, self.spectrum_fns.baseplane_rmsd, spectrum_id=data[1])
75
76
88
89
101
102
104 """The spectrum_id arg test of the spectrum.integration_points() user function."""
105
106
107 for data in DATA_TYPES:
108
109 if data[0] == 'str':
110 continue
111
112
113 self.assertRaises(RelaxStrError, self.spectrum_fns.integration_points, N=0, spectrum_id=data[1])
114
115
127
128
130 """The file arg test of the spectrum.read_intensities() user function."""
131
132
133 for data in DATA_TYPES:
134
135 if data[0] == 'str':
136 continue
137
138
139 self.assertRaises(RelaxStrListStrError, self.spectrum_fns.read_intensities, file=data[1])
140
141
143 """The dir arg test of the spectrum.read_intensities() user function."""
144
145
146 for data in DATA_TYPES:
147
148 if data[0] == 'None' or data[0] == 'str':
149 continue
150
151
152 self.assertRaises(RelaxNoneStrError, self.spectrum_fns.read_intensities, file='a', dir=data[1])
153
154
156 """The spectrum_id arg test of the spectrum.read_intensities() user function."""
157
158
159 for data in DATA_TYPES:
160
161 if data[0] == 'str':
162 continue
163
164
165 self.assertRaises(RelaxStrListStrError, self.spectrum_fns.read_intensities, spectrum_id=data[1])
166
167
169 """The dim arg test of the spectrum.read_intensities() user function."""
170
171
172 for data in DATA_TYPES:
173
174 if data[0] == 'int' or data[0] == 'bin':
175 continue
176
177
178 self.assertRaises(RelaxIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', dim=data[1])
179
180
182 """The int_col arg test of the spectrum.read_intensities() user function."""
183
184
185 for data in DATA_TYPES:
186
187 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin' or data[0] == 'int list':
188 continue
189
190
191 self.assertRaises(RelaxNoneIntListIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_col=data[1])
192
193
195 """The int_method arg test of the spectrum.read_intensities() user function."""
196
197
198 for data in DATA_TYPES:
199
200 if data[0] == 'str':
201 continue
202
203
204 self.assertRaises(RelaxStrError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method=data[1])
205
206
208 """The mol_name_col arg test of the spectrum.read_intensities() user function."""
209
210
211 for data in DATA_TYPES:
212
213 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin':
214 continue
215
216
217 self.assertRaises(RelaxNoneIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', mol_name_col=data[1])
218
219
221 """The res_num_col arg test of the spectrum.read_intensities() user function."""
222
223
224 for data in DATA_TYPES:
225
226 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin':
227 continue
228
229
230 self.assertRaises(RelaxNoneIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', res_num_col=data[1])
231
232
234 """The res_name_col arg test of the spectrum.read_intensities() user function."""
235
236
237 for data in DATA_TYPES:
238
239 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin':
240 continue
241
242
243 self.assertRaises(RelaxNoneIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', res_name_col=data[1])
244
245
247 """The spin_num_col arg test of the spectrum.read_intensities() user function."""
248
249
250 for data in DATA_TYPES:
251
252 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin':
253 continue
254
255
256 self.assertRaises(RelaxNoneIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', spin_num_col=data[1])
257
258
260 """The spin_name_col arg test of the spectrum.read_intensities() user function."""
261
262
263 for data in DATA_TYPES:
264
265 if data[0] == 'None' or data[0] == 'int' or data[0] == 'bin':
266 continue
267
268
269 self.assertRaises(RelaxNoneIntError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', spin_name_col=data[1])
270
271
273 """The sep arg test of the spectrum.read_intensities() user function."""
274
275
276 for data in DATA_TYPES:
277
278 if data[0] == 'None' or data[0] == 'str':
279 continue
280
281
282 self.assertRaises(RelaxNoneStrError, self.spectrum_fns.read_intensities, file='a', spectrum_id='x', int_method='y', sep=data[1])
283