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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import cmor
import numpy
import unittest
from base_CMIP6_CV import BaseCVsTest
def run():
unittest.main()
class TestNaNCheck(BaseCVsTest):
def test_nan_check(self):
cmor.setup(inpath='Tables',
netcdf_file_action=cmor.CMOR_REPLACE_4,
logfile=self.tmpfile)
cmor.dataset_json("Test/CMOR_input_example.json")
cmor.load_table("CMIP6_Amon.json")
itime = cmor.axis(table_entry='time',
units='days since 2000-01-01 00:00:00',
coord_vals=[15, 45, 75],
cell_bounds=[0, 30, 60, 90])
ilat = cmor.axis(table_entry='latitude',
units='degrees_north',
coord_vals=[0],
cell_bounds=[-1, 1])
ilon = cmor.axis(table_entry='longitude',
units='degrees_east',
coord_vals=[90],
cell_bounds=[89, 91])
axis_ids = [itime, ilat, ilon]
varid = cmor.variable('ts', 'K', axis_ids)
with self.assertRaises(cmor.CMORError):
_ = cmor.write(varid, [273, numpy.nan, 273])
self.assertCV(
"Invalid value(s) detected for variable 'ts' "
"(table: Amon): 1 values were NaNs."
)
self.assertCV(
"time: 1/45 lat: 0/0 lon: 0/90",
line_trigger="First encountered NaN was at (axis: index/value):"
)
def test_nan_check_with_basin_axis(self):
cmor.setup(inpath='Tables',
netcdf_file_action=cmor.CMOR_REPLACE_4,
logfile=self.tmpfile)
cmor.dataset_json("Test/CMOR_input_example.json")
cmor.load_table("CMIP6_Omon.json")
ntime = 12
nlat = 90
dlat = 180 / nlat
lats = numpy.arange(-90 + dlat / 2., 90, dlat)
blats = numpy.arange(-90, 90 + dlat, dlat)
ilat = cmor.axis(
table_entry='latitude',
coord_vals=lats,
cell_bounds=blats,
units='degrees_north')
myregions = [
"atlantic_ocean",
"indian_pacific_ocean",
"pacific_ocean",
"atlantic_arctic_ocean",
"global_ocean"]
nreg = len(myregions)
ireg = cmor.axis(table_entry='basin', units='1', coord_vals=myregions)
itim = cmor.axis(
table_entry='time',
units='months since 2030',
interval='1 month',
coord_vals=numpy.arange(ntime),
cell_bounds=numpy.arange(
ntime + 1))
varid = cmor.variable(table_entry='htovgyre', units='W',
axis_ids=numpy.array([itim, ireg, ilat]))
data = numpy.random.random((ntime, nreg, nlat)) * 3.E14
data[3, 2, 5] = numpy.nan
with self.assertRaises(cmor.CMORError):
_ = cmor.write(varid, data)
self.assertCV(
"Invalid value(s) detected for variable 'htovgyre' "
"(table: Omon): 1 values were NaNs."
)
self.assertCV(
"time: 3/105 basin: 2/pacific_ocean lat: 5/-79",
line_trigger="First encountered NaN was at (axis: index/value):"
)
if __name__ == '__main__':
run()
|
