def test_proper_subregion_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(
self.evaluation,
file_path=file_path
)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('subregions' in data)
first_bounds = [
self.subregions[0].lat_min,
self.subregions[0].lat_max,
self.subregions[0].lon_min,
self.subregions[0].lon_max,
]
second_bounds = [
self.subregions[1].lat_min,
self.subregions[1].lat_max,
self.subregions[1].lon_min,
self.subregions[1].lon_max,
]
self.assertEqual(first_bounds, data['subregions'][0])
self.assertEqual(second_bounds, data['subregions'][1])
def test_full_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(
self.evaluation,
file_path=file_path
)
self.assertTrue(os.path.isfile(file_path))
def test_proper_dataset_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(self.evaluation,
file_path=file_path)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('datasets' in data)
self.assertTrue('reference' in data['datasets'])
self.assertTrue('targets' in data['datasets'])
self.assertAlmostEqual(
writer.generate_dataset_information(self.evaluation),
data['datasets'])
def test_proper_evaluation_setting_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(self.evaluation,
file_path=file_path)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('evaluation' in data)
self.assertTrue('temporal_time_delta' in data['evaluation'])
self.assertTrue('spatial_regrid_lats' in data['evaluation'])
self.assertTrue('spatial_regrid_lons' in data['evaluation'])
self.assertTrue('subset' in data['evaluation'])
self.assertAlmostEqual(
writer.generate_evaluation_information(self.evaluation),
data['evaluation'])
def test_proper_dataset_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(
self.evaluation,
file_path=file_path
)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('datasets' in data)
self.assertTrue('reference' in data['datasets'])
self.assertTrue('targets' in data['datasets'])
self.assertAlmostEqual(
writer.generate_dataset_information(self.evaluation),
data['datasets']
)
def test_proper_evaluation_setting_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(
self.evaluation,
file_path=file_path
)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('evaluation' in data)
self.assertTrue('temporal_time_delta' in data['evaluation'])
self.assertTrue('spatial_regrid_lats' in data['evaluation'])
self.assertTrue('spatial_regrid_lons' in data['evaluation'])
self.assertTrue('subset' in data['evaluation'])
self.assertAlmostEqual(
writer.generate_evaluation_information(self.evaluation),
data['evaluation']
)
def test_proper_metric_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(self.evaluation,
file_path=file_path)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('metrics' in data)
self.assertTrue(type(data['metrics']) == type(list()))
for metric in self.evaluation.metrics:
self.assertTrue(metric.__class__.__name__ in data['metrics'])
for metric in self.evaluation.unary_metrics:
self.assertTrue(metric.__class__.__name__ in data['metrics'])
total_eval_metrics = (len(self.evaluation.metrics) +
len(self.evaluation.unary_metrics))
self.assertTrue(total_eval_metrics, len(data['metrics']))
def test_proper_metric_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(
self.evaluation,
file_path=file_path
)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('metrics' in data)
self.assertTrue(type(data['metrics']) == type(list()))
for metric in self.evaluation.metrics:
self.assertTrue(metric.__class__.__name__ in data['metrics'])
for metric in self.evaluation.unary_metrics:
self.assertTrue(metric.__class__.__name__ in data['metrics'])
total_eval_metrics = (
len(self.evaluation.metrics) +
len(self.evaluation.unary_metrics)
)
self.assertTrue(total_eval_metrics, len(data['metrics']))
def test_proper_subregion_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(self.evaluation,
file_path=file_path)
data = yaml.load(open(file_path, 'r'))
self.assertTrue('subregions' in data)
first_bounds = [
self.subregions[0].lat_min,
self.subregions[0].lat_max,
self.subregions[0].lon_min,
self.subregions[0].lon_max,
]
second_bounds = [
self.subregions[1].lat_min,
self.subregions[1].lat_max,
self.subregions[1].lon_min,
self.subregions[1].lon_max,
]
self.assertEqual(first_bounds, data['subregions'][0])
self.assertEqual(second_bounds, data['subregions'][1])
def test_full_export(self):
file_path = '/tmp/test_config.yaml'
writer.export_evaluation_to_config(self.evaluation,
file_path=file_path)
self.assertTrue(os.path.isfile(file_path))
