def test_missing_data_files(self):
"""Test with missing files."""
diagnostic_name = 'temperature_on_height_levels'
msg = 'The relevant data files for'
with self.assertRaisesRegexp(IOError, msg):
get_additional_diagnostics(diagnostic_name, 'not_a_valid_path')
def test_available_data_files_with_invalid_time_extraction(self):
"""Test with files available and an attempted extraction of data at a
time that is not valid."""
diagnostic_name = 'temperature_on_height_levels'
msg = 'No diagnostics match .*'
time_extract = Constraint(time=PartialDateTime(2018, 1, 1, 0))
with self.assertRaisesRegexp(ValueError, msg):
get_additional_diagnostics(diagnostic_name, self.data_directory,
time_extract=time_extract)
def test_available_data_files(self):
"""Test with files available."""
diagnostic_name = 'temperature_on_height_levels'
result = get_additional_diagnostics(diagnostic_name,
self.data_directory)
self.assertIsInstance(result, CubeList)
self.assertArrayEqual(result[0].data,
self.temperature_on_height_levels.data)
def test_available_data_files_with_time_extraction(self):
"""Test with files available and an extraction of data at a given
time."""
diagnostic_name = 'temperature_on_height_levels'
result = get_additional_diagnostics(diagnostic_name,
self.data_directory,
time_extract=self.time_extract)
self.assertIsInstance(result, CubeList)
self.assertArrayEqual(result[0].data,
self.temperature_on_height_levels[0].data)