Codebase list netcdf4-python / upstream/1.4.3.2 test / tst_get_variables_by_attributes.py
upstream/1.4.3.2

Tree @upstream/1.4.3.2 (Download .tar.gz)

tst_get_variables_by_attributes.py @upstream/1.4.3.2 — raw · history · blame 

 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
import os
import unittest

import netCDF4

class VariablesByAttributesTests(unittest.TestCase):

    def setUp(self):
        netcdf_file = os.path.join(os.path.dirname(__file__), "netcdf_dummy_file.nc")
        self.nc = netCDF4.Dataset(netcdf_file)

    def test_find_variables_by_single_attribute(self):
        vs = self.nc.get_variables_by_attributes(axis='Z')
        self.assertEqual(len(vs), 1)

        vs = self.nc.get_variables_by_attributes(units='m/s')
        self.assertEqual(len(vs), 4)

    def test_find_variables_by_multiple_attribute(self):
        vs = self.nc.get_variables_by_attributes(axis='Z', units='m')
        self.assertEqual(len(vs), 1)

    def test_find_variables_by_single_lambda(self):
        vs = self.nc.get_variables_by_attributes(axis=lambda v: v in ['X', 'Y', 'Z', 'T'])
        self.assertEqual(len(vs), 1)

        vs = self.nc.get_variables_by_attributes(grid_mapping=lambda v: v is not None)
        self.assertEqual(len(vs), 12)

    def test_find_variables_by_multiple_lambdas(self):
        vs = self.nc.get_variables_by_attributes(grid_mapping=lambda v: v is not None,
                                                 long_name=lambda v: v is not None and 'Upward (w) velocity' in v)
        self.assertEqual(len(vs), 1)

    def test_find_variables_by_attribute_and_lambda(self):
        vs = self.nc.get_variables_by_attributes(units='m/s',
                                                 grid_mapping=lambda v: v is not None)
        self.assertEqual(len(vs), 4)

        vs = self.nc.get_variables_by_attributes(grid_mapping=lambda v: v is not None,
                                                 long_name='Upward (w) velocity')
        self.assertEqual(len(vs), 1)

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