Newer
Older
# coding=utf-8
from unittest import TestCase
from earthdiagnostics.config import CMORConfig, ConfigException, THREDDSConfig, ReportConfig, ExperimentConfig
from earthdiagnostics.frequency import Frequencies
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from earthdiagnostics.modelingrealm import ModelingRealms
class VariableMock(object):
def __init__(self):
self.domain = ModelingRealms.ocean
self.short_name = 'tos'
def __eq__(self, other):
return self.domain == other.domain and self.short_name == other.short_name
class VariableManagerMock(object):
def get_variable(self, alias, silent=False):
if alias == 'bad':
return None
var = VariableMock()
var.short_name = alias
return var
class ParserMock(object):
def __init__(self):
self._values = {}
def add_value(self, section, var, value):
self._values[self.get_var_string(section, var)] = value
def get_var_string(self, section, var):
return '{0}:{1}'.format(section, var)
def get_value(self, section, var, default):
try:
return self._values[self.get_var_string(section, var)]
except KeyError:
return default
def get_bool_option(self, section, var, default):
return self.get_value(section, var, default)
def get_path_option(self, section, var, default):
return self.get_value(section, var, default)
def get_int_option(self, section, var, default=0):
return self.get_value(section, var, default)
def get_int_list_option(self, section, var, default=list(), separator=' '):
try:
return [int(val) for val in self._values[self.get_var_string(section, var)].split(separator)]
except KeyError:
return default
def get_list_option(self, section, var, default=list(), separator=' '):
try:
return [val for val in self._values[self.get_var_string(section, var)].split(separator)]
except KeyError:
return default
def get_option(self, section, var, default=None):
return self.get_value(section, var, default)
class TestCMORConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
self.var_manager = VariableManagerMock()
def test_basic_config(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.ocean, True)
self.assertEquals(config.atmosphere, True)
self.assertEquals(config.force, False)
self.assertEquals(config.force_untar, False)
self.assertEquals(config.use_grib, True)
self.assertEquals(config.activity, 'CMIP')
self.assertEquals(config.associated_experiment, 'to be filled')
self.assertEquals(config.associated_model, 'to be filled')
self.assertEquals(config.initialization_description, 'to be filled')
self.assertEquals(config.initialization_method, '1')
self.assertEquals(config.initialization_number, 1)
self.assertEquals(config.source, 'to be filled')
self.assertEquals(config.version, '')
self.assertEquals(config.physics_version, '1')
self.assertEquals(config.physics_description, 'to be filled')
self.assertEquals(config.filter_files, '')
self.assertEquals(config.default_atmos_grid, 'gr')
self.assertEquals(config.default_ocean_grid, 'gn')
def test_cmorize(self):
config = CMORConfig(self.mock_parser, self.var_manager)
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
self.assertTrue(config.cmorize(VariableMock()))
self.assertTrue(config.cmorize(None))
def test_cmorize_list(self):
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'ocean:thetao ocean:tos')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.cmorize(VariableMock()))
thetao_mock = VariableMock()
thetao_mock.domain = ModelingRealms.ocean
thetao_mock.short_name = 'thetao'
self.assertTrue(config.cmorize(thetao_mock))
def test_bad_list(self):
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', '#ocean:tos')
with self.assertRaises(ConfigException):
CMORConfig(self.mock_parser, self.var_manager)
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'atmos:tos')
with self.assertRaises(ConfigException):
CMORConfig(self.mock_parser, self.var_manager)
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'ocean:bad')
with self.assertRaises(ConfigException):
CMORConfig(self.mock_parser, self.var_manager)
def test_not_cmorize(self):
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'ocean:tos')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.cmorize(VariableMock()))
self.assertFalse(config.cmorize(None))
tas_mock = VariableMock()
tas_mock.domain = ModelingRealms.atmos
tas_mock.short_name = 'tas'
self.assertFalse(config.cmorize(tas_mock))
thetao_mock = VariableMock()
thetao_mock.domain = ModelingRealms.ocean
thetao_mock.short_name = 'thetao'
self.assertFalse(config.cmorize(thetao_mock))
def test_comment(self):
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'ocean:tos #ocean:thetao ')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.cmorize(VariableMock()))
thetao_mock = VariableMock()
thetao_mock.domain = ModelingRealms.ocean
thetao_mock.short_name = 'thetao'
self.assertFalse(config.cmorize(thetao_mock))
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', '#ocean:tos ocean:thetao ')
with self.assertRaises(ConfigException):
CMORConfig(self.mock_parser, self.var_manager)
def test_cmorization_chunk(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.chunk_cmorization_requested(1))
def test_cmorize_only_some_chunks(self):
self.mock_parser.add_value('CMOR', 'CHUNKS', '3 5')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.chunk_cmorization_requested(3))
self.assertTrue(config.chunk_cmorization_requested(5))
self.assertFalse(config.chunk_cmorization_requested(1))
self.assertFalse(config.chunk_cmorization_requested(4))
self.assertFalse(config.chunk_cmorization_requested(6))
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
def test_any_required(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.any_required(['tos']))
self.mock_parser.add_value('CMOR', 'VARIABLE_LIST', 'ocean:tos ocean:thetao')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertTrue(config.any_required(['tos', 'thetao', 'tas']))
self.assertTrue(config.any_required(['tos', 'tas']))
self.assertTrue(config.any_required(['thetao']))
self.assertFalse(config.any_required(['tas']))
def test_hourly_vars(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.six_hourly), {})
self.mock_parser.add_value('CMOR', 'ATMOS_HOURLY_VARS', '128,129:1,130:1-2,131:1:10,132:0:10:5')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.six_hourly), {128: None,
129: '1',
130: '1,2',
131: '1,2,3,4,5,6,7,8,9',
132: '0,5'})
self.assertEquals(config.get_levels(Frequencies.six_hourly, 128), None)
self.assertEquals(config.get_levels(Frequencies.six_hourly, 129), '1')
self.assertEquals(config.get_levels(Frequencies.six_hourly, 130), '1,2')
self.assertEquals(config.get_levels(Frequencies.six_hourly, 131), '1,2,3,4,5,6,7,8,9',)
self.assertEquals(config.get_levels(Frequencies.six_hourly, 132), '0,5')
def test_daily_vars(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.daily), {})
self.mock_parser.add_value('CMOR', 'ATMOS_DAILY_VARS', '128,129:1,130:1-2,131:1:10,132:0:10:5')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.daily), {128: None,
129: '1',
130: '1,2',
131: '1,2,3,4,5,6,7,8,9',
132: '0,5'})
self.assertEquals(config.get_levels(Frequencies.daily, 128), None)
self.assertEquals(config.get_levels(Frequencies.daily, 129), '1')
self.assertEquals(config.get_levels(Frequencies.daily, 130), '1,2')
self.assertEquals(config.get_levels(Frequencies.daily, 131), '1,2,3,4,5,6,7,8,9',)
self.assertEquals(config.get_levels(Frequencies.daily, 132), '0,5')
def test_monthly_vars(self):
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.monthly), {})
self.mock_parser.add_value('CMOR', 'ATMOS_MONTHLY_VARS', '128,129:1,130:1-2,131:1:10,132:0:10:5')
config = CMORConfig(self.mock_parser, self.var_manager)
self.assertEquals(config.get_variables(Frequencies.monthly), {128: None,
129: '1',
130: '1,2',
131: '1,2,3,4,5,6,7,8,9',
132: '0,5'})
self.assertEquals(config.get_levels(Frequencies.monthly, 128), None)
self.assertEquals(config.get_levels(Frequencies.monthly, 129), '1')
self.assertEquals(config.get_levels(Frequencies.monthly, 130), '1,2')
self.assertEquals(config.get_levels(Frequencies.monthly, 131), '1,2,3,4,5,6,7,8,9',)
self.assertEquals(config.get_levels(Frequencies.monthly, 132), '0,5')
def test_bad_frequency_vars(self):
config = CMORConfig(self.mock_parser, self.var_manager)
with self.assertRaises(ValueError):
self.assertEquals(config.get_variables(Frequencies.climatology), {})
class TestTHREDDSConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
def test_basic_config(self):
config = THREDDSConfig(self.mock_parser)
self.assertEquals(config.server_url, '')
def test_url(self):
self.mock_parser.add_value('THREDDS', 'SERVER_URL', 'test_url')
config = THREDDSConfig(self.mock_parser)
self.assertEquals(config.server_url, 'test_url')
class TestReportConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
def test_basic_config(self):
config = ReportConfig(self.mock_parser)
self.assertEquals(config.path, '')
self.assertEquals(config.maximum_priority, 10)
def test_path(self):
self.mock_parser.add_value('REPORT', 'PATH', 'new_path')
config = ReportConfig(self.mock_parser)
self.assertEquals(config.path, 'new_path')
def test_priority(self):
self.mock_parser.add_value('REPORT', 'MAXIMUM_PRIORITY', 3)
config = ReportConfig(self.mock_parser)
self.assertEquals(config.maximum_priority, 3)
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
class TestExperimentConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
def test_basic_config(self):
config = ExperimentConfig(self.mock_parser)
self.assertEquals(config.startdates, [])
self.assertEquals(config.members, [])
self.assertEquals(config.chunk_size, 0)
self.assertEquals(config.num_chunks, 0)
self.assertEquals(config.atmos_grid, '')
self.assertEquals(config.atmos_timestep, 6)
self.assertEquals(config.ocean_timestep, 6)
def test_startdates(self):
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '20001101 20011101')
config = ExperimentConfig(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20011101'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '200(0|1)1101')
config = ExperimentConfig(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20011101'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '200[0-2](02|05|08|11)01')
config = ExperimentConfig(self.mock_parser)
print(config.startdates)
self.assertEquals(config.startdates, [u'20000201', u'20000501', u'20000801', u'20001101', u'20010201',
u'20010501', u'20010801', u'20011101', u'20020201', u'20020501',
u'20020801', u'20021101'])