File: test_aggregate.py

package info (click to toggle)
cf-python 1.3.2+dfsg1-4
  • links: PTS, VCS
  • area: main
  • in suites: sid, stretch
  • size: 7,996 kB
  • sloc: python: 51,733; ansic: 2,736; makefile: 78; sh: 2
file content (71 lines) | stat: -rw-r--r-- 2,736 bytes parent folder | download
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)