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
|
import cf
import numpy
import os
import unittest
class CoordinateReferenceTest(unittest.TestCase):
filename = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'test_file.nc')
chunk_sizes = (17, 34, 300, 100000)[::-1]
def test_CoordinateReference_equals(self):
f = cf.read(self.filename)[0]
t = cf.CoordinateReference(name='atmosphere_hybrid_height_coordinate',
a='aux0', b='aux1', orog=f,
coord_terms=('a', 'b'))
# print t.dump(complete=True)
# print t
# t.inspect()
# print
self.assertTrue(t.equals(t.copy(), traceback=True))
# Create a rotated_latitude_longitude grid mapping coordinate
# reference
t = cf.CoordinateReference(name='rotated_latitude_longitude',
grid_north_pole_latitude=38.0,
grid_north_pole_longitude=190.0)
# print t.dump(complete=True)
# print t
# t.inspect()
# print
self.assertTrue(t.equals(t.copy(), traceback=True))
#--- End: def
def test_Field_ref_refs(self):
f = cf.read(self.filename)[0]
self.assertTrue(f.ref('BLAH') is None)
self.assertTrue(f.ref('atmos', key=True) == 'ref0')
self.assertTrue(f.ref('atmos', key=True, inverse=True) == 'ref1')
self.assertTrue(set(f.refs()) == set(['ref0', 'ref1']))
self.assertTrue(set(f.refs('BLAH')) == set())
self.assertTrue(set(f.refs('rot')) == set(['ref1']))
self.assertTrue(set(f.refs('rot', inverse=True)) == set(['ref0']))
self.assertTrue(set(f.refs('atmosphere_hybrid_height_coordinate', exact=True)) == set(['ref0']))
#--- End: def
#--- End: class
if __name__ == '__main__':
print 'cf-python version:', cf.__version__
print 'cf-python path:' , os.path.abspath(cf.__file__)
print ''
unittest.main(verbosity=2)
|
