def test_fix_metadata(self):
"""Test data fix."""
fix = Sit(self.var_info_mock)
cube = fix.fix_metadata((self.cube, ))[0]
time = cube.coord('time')
self.assertEqual(time.bounds[0, 0], 44984)
self.assertEqual(time.bounds[0, 1], 45015)
self.assertEqual(time.bounds[1, 0], 45015)
self.assertEqual(time.bounds[1, 1], 45045)
def test_fix_metadata_day_do_nothing(self):
"""Test data fix."""
self.var_info_mock.frequency = 'day'
fix = Sit(self.var_info_mock)
cube = fix.fix_metadata((self.cube, ))[0]
time = cube.coord('time')
self.assertEqual(time.bounds[0, 0], 1e8)
self.assertEqual(time.bounds[0, 1], 1.1e8)
self.assertEqual(time.bounds[1, 0], 1.1e8)
self.assertEqual(time.bounds[1, 1], 1.2e8)
def test_fix_metadata_not_needed(self):
"""Test data fix."""
fix = Sit(self.var_info_mock)
cube = fix.fix_metadata((self.cube, ))[0]
time = cube.coord('time')
new_bounds = [[44985., 45014.], [45016., 45044.]]
time.bounds = new_bounds
self.assertEqual(time.bounds[0, 0], 44985)
self.assertEqual(time.bounds[0, 1], 45014)
self.assertEqual(time.bounds[1, 0], 45016)
self.assertEqual(time.bounds[1, 1], 45044)
def setUp(self):
"""Prepare tests."""
self.cube = Cube([1.0, 2.0], var_name='sit', units='m')
self.cube.add_dim_coord(
iris.coords.DimCoord(points=[45000.5, 45031.5],
var_name='time',
standard_name='time',
long_name='time',
units='days since 1850-01-01',
bounds=[[1e8, 1.1e8], [1.1e8, 1.2e8]]), 0)
self.var_info_mock = mock.Mock()
self.var_info_mock.frequency = 'mon'
self.var_info_mock.short_name = 'sit'
self.fix = Sit(self.var_info_mock)
def test_get(self):
"""Test fix get"""
self.assertListEqual(
Fix.get_fixes('CMIP5', 'GFDL-CM2P1', 'OImon', 'sit'),
[Sit(self.var_info_mock), AllVars(None)])