-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_CMIP6_albedo.py
92 lines (72 loc) · 3.88 KB
/
test_CMIP6_albedo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import unittest
import CMIP6_light
import CMIP6_config
import CMIP6_model
import CMIP6_IO
from datetime import datetime
import numpy as np
import xarray as xr
# Unittest for ``CMIP6_light` setup
class TestCMIP6_light(unittest.TestCase):
def setUp(self):
self.cmip6 = CMIP6_light.CMIP6_light()
self.cmip6_model=CMIP6_model.CMIP6_MODEL(name="test")
self.cmip6_IO = CMIP6_IO.CMIP6_IO()
self.query_string = "source_id=='ACCESS-ESM1-5'and table_id=='Amon' and grid_label=='gn' \
and experiment_id=='historical' and variable_id=='uas'"
class TestMethods(TestCMIP6_light):
def test_create_data_array(self):
self.assertIsNotNone(self.cmip6)
self.assertIsNotNone(self.cmip6.config)
class TestIO(TestCMIP6_light):
def test_query_returns_dataset_with_timesteps(self):
# We avoid calling the read_cmip6_repository in other tests as it is time consuming
try:
self.cmip6.config.df
except:
self.cmip6.config.read_cmip6_repository()
ds_hist = self.cmip6_IO.perform_cmip6_query(self.cmip6.config, self.query_string)
self.assertIsNotNone(ds_hist)
self.assertIsInstance(ds_hist, xr.Dataset)
self.assertTrue(len(ds_hist["time"].values) > 0)
class TestModel(TestCMIP6_light):
def test_access_to_variable_id_using_keys(self):
# Make sure dictionary is empty at init
self.assertFalse(self.cmip6_model.ds_sets)
class TestInit(TestCMIP6_light):
def test_verify_equal_length_variable_and_table_ids(self):
self.assertEqual(len(self.cmip6.config.variable_ids), len(self.cmip6.config.table_ids))
def test_setup_parameters(self):
self.cmip6.config.setup_parameters()
self.assertIsNotNone(self.cmip6.config.wavelengths)
self.assertIsNotNone(self.cmip6.config.alpha_chl)
self.assertIsNotNone(self.cmip6.config.alpha_w)
self.assertIsNotNone(self.cmip6.config.beta_w)
self.assertIsNotNone(self.cmip6.config.alpha_wc)
self.assertIsNotNone(self.cmip6.config.solar_energy)
def test_shape_of_arrays_equal_after_setup_parameters(self):
self.cmip6.config.setup_parameters()
self.assertEqual(np.shape(self.cmip6.config.wavelengths), np.shape(self.cmip6.config.alpha_chl))
self.assertEqual(np.shape(self.cmip6.config.wavelengths), np.shape(self.cmip6.config.alpha_w))
self.assertEqual(np.shape(self.cmip6.config.wavelengths), np.shape(self.cmip6.config.beta_w))
self.assertEqual(np.shape(self.cmip6.config.wavelengths), np.shape(self.cmip6.config.alpha_wc))
self.assertEqual(np.shape(self.cmip6.config.wavelengths), np.shape(self.cmip6.config.solar_energy))
def test_initial_models_empty(self):
self.assertFalse(self.cmip6.cmip6_models)
def test_initial_config_not_null(self):
self.assertIsNotNone(self.cmip6.config)
def test_initial_start_and_end_dates(self):
self.assertIsNotNone(self.cmip6.config.start_date)
self.assertIsNotNone(self.cmip6.config.end_date)
def test_inital_start_and_end_dates_correct_format(self):
start_date = datetime.strptime(self.cmip6.config.start_date, '%Y-%m-%d')
end_date = datetime.strptime(self.cmip6.config.end_date, '%Y-%m-%d')
self.assertIsInstance(start_date, datetime,
"Make sure that start and end date in `CMIP6_config` is of correct format `YYYY-mm-dd`")
self.assertIsInstance(end_date, datetime,
"Make sure that start and end date in `CMIP6_config` is of correct format `YYYY-mm-dd`")
def test_initial_variable_and_table_ids_equal_length(self):
self.assertTrue(len(self.cmip6.config.table_ids) == len(self.cmip6.config.variable_ids),
"Make sure that you initialize CMIP6_config using equal length of variable ids and table ids")
if __name__ == "__main__":
unittest.main()