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
|
import cf
import os
import unittest
class aggregateTest(unittest.TestCase):
filename = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'test_file.nc')
chunk_sizes = (17, 34, 300, 100000)[::-1]
original_chunksize = cf.CHUNKSIZE()
def test_aggregate(self):
for chunksize in self.chunk_sizes:
cf.CHUNKSIZE(chunksize)
f = cf.read(self.filename, squeeze=True)[0]
f.aux('aux0').id = 'atmosphere_hybrid_height_coordinate_ak'
f.aux('aux1').id = 'atmosphere_hybrid_height_coordinate_bk'
g = cf.FieldList(f.subspace[0])
g.append(f.subspace[1:3])
g.append(f.subspace[3])
g[-1].flip(0, i=True)
g.append(f.subspace[4:7])
g[-1].flip(0, i=True)
g.extend([f.subspace[i] for i in range(7, f.shape[0])])
g0 = g.copy()
self.assertTrue(g.equals(g0, traceback=True), "g != itself")
h = cf.aggregate(g, info=1)
self.assertTrue(h[0].shape == (10, 9), 'h[0].shape is '+repr(h[0].shape))
self.assertTrue(h[0].equals(f, traceback=True), 'asdasds')
self.assertTrue(g.equals(g0, traceback=True), 'g != itself after aggregation')
i = cf.aggregate(g, info=1)
self.assertTrue(i.equals(h, traceback=True), 'The second aggregation != the first')
self.assertTrue(g.equals(g0, traceback=True), 'g != itself after the second aggregation')
i = cf.aggregate(g, info=1, axes='grid_latitude')
self.assertTrue(i.equals(h, traceback=True), 'The third aggregation != the first')
self.assertTrue(g.equals(g0, traceback=True), 'g !=itself after the third aggregation')
self.assertTrue(i[0].shape == (10,9), 'i[0].shape is '+repr(i[0].shape))
i = cf.aggregate(g, info=1, axes='grid_latitude', donotchecknonaggregatingaxes=1)
self.assertTrue(i.equals(h, traceback=True), 'The fourth aggregation != the first')
self.assertTrue(g.equals(g0, traceback=True), 'g != itself after the fourth aggregation')
self.assertTrue(i[0].shape == (10,9), 'i[0].shape is '+repr(i[0].shape))
#--- End: for
cf.CHUNKSIZE(self.original_chunksize)
#--- 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)
|
