def setUp(self):
"""Prepare for testing"""
self.cube_1 = mock.Mock()
self.cube_1.var_name = 'cube1'
self.cube_2 = mock.Mock()
self.cube_2.var_name = 'cube2'
self.cubes = [self.cube_1, self.cube_2]
self.fix = Fix()
def setUp(self):
"""Prepare for testing."""
self.cube_1 = Mock()
self.cube_1.var_name = 'cube1'
self.cube_2 = Mock()
self.cube_2.var_name = 'cube2'
self.cubes = [self.cube_1, self.cube_2]
vardef = Mock()
vardef.short_name = 'fix'
self.fix = Fix(vardef)
def test_fixed_filenam(self):
filepath = os.path.join(self.temp_folder, 'file.nc')
output_dir = os.path.join(self.temp_folder, 'fixed')
os.makedirs(output_dir)
fixed_filepath = Fix(None).get_fixed_filepath(output_dir, filepath)
self.assertTrue(fixed_filepath, os.path.join(output_dir, 'file.nc'))
def test_fix_file(self):
filepath = 'sample_filepath'
self.assertEqual(Fix(None).fix_file(filepath, 'preproc'), filepath)
def test_fix_data(self):
cube = Cube([0])
reference = Cube([0])
self.assertEqual(Fix(None).fix_data(cube), reference)