File: test_cmor_CMIP6Plus.py

package info (click to toggle)
cmor 3.13.2-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 16,960 kB
  • sloc: ansic: 28,094; f90: 13,872; python: 12,423; sh: 3,738; makefile: 111
file content (64 lines) | stat: -rw-r--r-- 1,942 bytes parent folder | download | duplicates (2)
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
import cmor
import numpy
import unittest


class TestCMIP6Plus(unittest.TestCase):

    def test_multiple_modeling_realms(self):
        ntimes_passed = 12

        self.assertEqual(
            cmor.setup(inpath='mip-cmor-tables/Tables',
                       netcdf_file_action=cmor.CMOR_REPLACE),
            0)
        self.assertEqual(cmor.dataset_json("Test/CMIP6Plus_user_input.json"),
                         0)

        self.assertEqual(cmor.load_table("MIP_OPmon.json"), 0)

        axes = [
                {
                    'table_entry': 'time',
                    'units': 'months since 2010-1-1',
                    'coord_vals': numpy.arange(0, ntimes_passed, 1),
                    'cell_bounds': numpy.arange(0, ntimes_passed+1, 1)
                },
                {
                    'table_entry': 'latitude',
                    'units': 'degrees_north',
                    'coord_vals': [0],
                    'cell_bounds': [-1, 1]
                },
                {
                    'table_entry': 'longitude',
                    'units': 'degrees_east',
                    'coord_vals': [90],
                    'cell_bounds': [89, 91]
                }
            ]

        axis_ids = [cmor.axis(**axis) for axis in axes]

        ivar = cmor.variable('hfsifrazil2d', units='W m-2', axis_ids=axis_ids)

        data = [280., ] * ntimes_passed

        self.assertEqual(
            cmor.write(ivar, data, ntimes_passed=ntimes_passed),
            0)

        self.assertFalse(cmor.has_cur_dataset_attribute('further_info_url'))

        realms = cmor.get_cur_dataset_attribute('realm')
        self.assertEqual(realms, "ocean seaIce")

        title = cmor.get_cur_dataset_attribute('title')
        self.assertEqual(title,
                         'HadGEM3-GC31-LL output prepared for CMIP6Plus')

        self.assertEqual(cmor.close(), 0)


if __name__ == '__main__':
    unittest.main()