Newer
Older
# coding=utf-8
from unittest import TestCase
from earthdiagnostics.config import CMORConfig, ConfigException, THREDDSConfig, ReportConfig, ExperimentConfig, Config
from earthdiagnostics.frequency import Frequencies
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(mock.Mock):
def __init__(self, **kwargs):
super().__init__(**kwargs)
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_choice_option(self, section, var, choices, default, ignore_case=True):
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)
def has_section(self, section):
start = '{0}:'.format(section)
return any(x.startswith(start) for x in self._values)
def options(self, section):
start = '{0}:'.format(section)
return [x[len(start):] for x in self._values if x.startswith(start)]
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')
self.assertEquals(config.min_cmorized_vars, 10)
def test_cmorize(self):
config = CMORConfig(self.mock_parser, self.var_manager)
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
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))
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
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):
config.get_variables(Frequencies.climatology)
def test_requested_codes(self):
self.mock_parser.add_value('CMOR', 'ATMOS_HOURLY_VARS', '128,129:1,130:1-2,131:1:10,132:0:10:5')
self.mock_parser.add_value('CMOR', 'ATMOS_DAILY_VARS', '128,129:1,130:1-2,131:1:10,132:0:10:5')
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_requested_codes(), {128, 129, 130, 131, 132})
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
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)
class TestExperimentConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
def test_basic_config(self):
config = ExperimentConfig()
config.parse_ini(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_members(self):
self.mock_parser.add_value('EXPERIMENT', 'MEMBERS', 'fc0 1')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.members, [0, 1])
self.mock_parser.add_value('EXPERIMENT', 'MEMBERS', 'fc00 fc01')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.members, [0, 1])
self.mock_parser.add_value('EXPERIMENT', 'MEMBERS', 'fc1-fc3')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.members, [1, 2, 3])
self.mock_parser.add_value('EXPERIMENT', 'MEMBERS', '1-3')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.members, [1, 2, 3])
def test_startdates(self):
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '20001101 20011101')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20011101'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '200(0|1)1101')
config = ExperimentConfig()
config.parse_ini(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()
config.parse_ini(self.mock_parser)
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'])
def test_auto_startdates(self):
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20011101,1Y}')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20011101'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20011101,6M} ')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20010501', '20011101'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20001201,1W}')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20001108', '20001115', '20001122', '20001129'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20001201,W}')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20001108', '20001115', '20001122', '20001129'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20001201,7D}')
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEquals(config.startdates, ['20001101', '20001108', '20001115', '20001122', '20001129'])
self.mock_parser.add_value('EXPERIMENT', 'STARTDATES', '{20001101,20001201,7F}')
with self.assertRaises(ConfigException):
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
def test_get_member_str(self):
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_member_str(1), 'fc1')
def test_get_full_years(self):
self.mock_parser.add_value('EXPERIMENT', 'CHUNK_SIZE', 3)
self.mock_parser.add_value('EXPERIMENT', 'CHUNKS', 15)
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_full_years('20000601'), [2001, 2002, 2003, 2004])
self.assertEqual(config.get_full_years('20000101'), [2000, 2001, 2002, 2003])
def test_get_year_chunks(self):
self.mock_parser.add_value('EXPERIMENT', 'CHUNK_SIZE', 3)
self.mock_parser.add_value('EXPERIMENT', 'CHUNKS', 15)
config = ExperimentConfig()
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_year_chunks('20000601', 2001), [3, 4, 5, 6, 7])
self.assertEqual(config.get_year_chunks('20000101', 2000), [1, 2, 3, 4])
self.assertEqual(config.get_year_chunks('20000601', 2000), [1, 2, 3])
self.assertEqual(config.get_year_chunks('20000601', 1999), [])
def test_get_chunk_list(self):
config = ExperimentConfig()
config.startdates = ('20010101', )
config.members = (0, 1, 2)
config.chunk_list = [0]
config.num_chunks = 2
self.assertEqual(config.get_chunk_list(), [('20010101', 0, 0), ('20010101', 1, 0), ('20010101', 2, 0)])
config.chunk_list = []
self.assertEqual(config.get_chunk_list(), [('20010101', 0, 1), ('20010101', 0, 2), ('20010101', 1, 1),
('20010101', 1, 2), ('20010101', 2, 1), ('20010101', 2, 2)])
def test_get_member_list(self):
config = ExperimentConfig()
config.startdates = ('20010101', )
config.members = (0, 1, 2)
self.assertEqual(config.get_member_list(), [('20010101', 0), ('20010101', 1), ('20010101', 2)])
def test_get_chunk_start_str(self):
config = ExperimentConfig()
self.mock_parser.add_value('EXPERIMENT', 'CHUNK_SIZE', 12)
self.mock_parser.add_value('EXPERIMENT', 'CHUNKS', 3)
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_chunk_start_str('20001101', 3), '20021101')
def test_get_chunk_start_str_datetime(self):
config = ExperimentConfig()
self.mock_parser.add_value('EXPERIMENT', 'CHUNK_SIZE', 12)
self.mock_parser.add_value('EXPERIMENT', 'CHUNKS', 3)
date = datetime.datetime(year=2000, month=11, day=1)
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_chunk_start_str(date, 3), '20021101')
def test_get_chunk_end_str(self):
config = ExperimentConfig()
self.mock_parser.add_value('EXPERIMENT', 'CHUNK_SIZE', 12)
self.mock_parser.add_value('EXPERIMENT', 'CHUNKS', 3)
config.parse_ini(self.mock_parser)
self.assertEqual(config.get_chunk_end_str('20001101', 3), '20031101')
class TestConfig(TestCase):
def setUp(self):
self.mock_parser = ParserMock()
self.mock_parser.add_value('DIAGNOSTICS', 'FREQUENCY', 'mon')
self.mock_parser.add_value('DIAGNOSTICS', 'DIAGS', 'diag1 diag2')
self.mock_parser.add_value('DIAGNOSTICS', 'SCRATCH_DIR', 'scratch')
self._environ = dict(os.environ)
def tearDown(self):
os.environ.clear()
os.environ.update(self._environ)
def mock_new():
return self.mock_parser
def mock_new_exp():
mock_exp = mock.Mock()
mock_exp.expid = 'expid'
return mock_exp
with mock.patch('earthdiagnostics.config.ConfigParser', new=mock_new):
with mock.patch('earthdiagnostics.config.VariableManager'):
with mock.patch('earthdiagnostics.config.ExperimentConfig', new=mock_new_exp):
with mock.patch('earthdiagnostics.config.CMORConfig'):
with mock.patch('earthdiagnostics.config.THREDDSConfig'):
config.parse('path')
def test_diags(self):
config = Config()
self.mock_parser.add_value('DIAGNOSTICS', 'DIAGS', 'diag1 diag2,opt1,opt2 # Commented diag')
self._parse(config)
self.assertEqual(config.get_commands(), (['diag1', 'diag2,opt1,opt2']))
def test_parse(self):
config = Config()
self._parse(config)
self.assertEqual(config.frequency, Frequencies.monthly)
self.assertEqual(config.auto_clean, True)
self.assertEqual(config.cdftools_path, '')
self.assertEqual(config.con_files, '')
self.assertEqual(config.data_adaptor, 'CMOR')
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
self.assertEqual(config.get_commands(), (['diag1', 'diag2']))
def test_alias(self):
config = Config()
self.mock_parser.add_value('ALIAS', 'diag1', 'diag3')
self._parse(config)
self.assertEqual(config.get_commands(), ['diag3', 'diag2'])
def test_auto_clean_ram_disk(self):
config = Config()
self.mock_parser.add_value('DIAGNOSTICS', 'AUTO_CLEAN', False)
self.mock_parser.add_value('DIAGNOSTICS', 'USE_RAMDISK', True)
self._parse(config)
self.assertEqual(config.auto_clean, True)
self.assertEqual(config.use_ramdisk, True)
def test_data_convention_primavera(self):
config = Config()
self.mock_parser.add_value('DIAGNOSTICS', 'DATA_CONVENTION', 'primavera')
self._parse(config)
self.assertEqual(config.data_convention, 'primavera')
self.assertEqual(config.scratch_masks, '/scratch/Earth/ocean_masks/primavera')
namelist = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..',
'earthdiagnostics/CDFTOOLS_primavera.namlist'))
self.assertEqual(os.environ['NAM_CDF_NAMES'], namelist)